题意(紫书P110):
输入n,得到编号为0~n-1的木块,分别摆放在顺序排列编号为0~n-1的位置。现对这些木块进行操作,操作分为四种。
1、move a onto b:把木块a、b上的木块放回各自的原位,再把a放到b上;
2、move a over b:把a上的木块放回各自的原位,再把a发到含b的堆上;
3、pile a onto b:把b上的木块放回各自的原位,再把a连同a上的木块移到b上;
4、pile a over b:把a连同a上木块移到含b的堆上。
当输入quit时,结束操作并输出0~n-1的位置上的木块情况
题解:
#include<iostream>
#include<vector>
using namespace std;
vector<int>s[30];
int n;
void find_dig(int a,int &p,int &h)
{
for(p=0;p<n;p++)//双重循环确定木块a的位置(高度以及堆的编号)
{
for(h=0;h<s[p].size();h++)
{
if(s[p][h]==a)
return ;
}
}
}
void clear_dig(int p,int h)
{
for(int i=h+1;i<s[p].size();i++)//将位置p上,高度为h以上的木块归位
{
int b=s[p][i];
s[b].push_back(b);
}
s[p].resize(h+1);//收缩s[p]的大小(即0~h)
}
void pile_dig(int pa,int ha,int pb)//将木块a以及以上的木块移到木块b上
{
for(int i=ha;i<s[pa].size();i++)
{
s[pb].push_back(s[pa][i]);
}
s[pa].resize(ha);//收缩s[pa]的大小(即ha)
}
int main()
{
//int n;
string s1,s2;
int a,b,pa,ha,pb,hb;
cin>>n;
for(int i=0;i<n;i++)
{
s[i].push_back(i);//将每个木块放到所对应的编号位置上,即<1,1>,<2,2>...
}
while(cin>>s1)
{
if(s1=="quit")
{
break;
}
cin>>a>>s2>>b;
find_dig(a,pa,ha);//找到编号a的木块的位置,并且用pa记录此时木块a所在的位置编号,以及木块a在pa位置的高度为ha
find_dig(b,pb,hb);//找到编号b的木块的位置,并且用pb记录此时木块b所在的位置编号,以及木块b在pb位置的高度为hb
if(pa==pb)//如果木块a和木块b在同一位置上,则为非法指令
{
continue;
}
if(s2=="onto")
{
clear_dig(pb,hb);//将位置b上的高度为hb的木块以上的木块进行归位
}
if(s1=="move")
{
clear_dig(pa,ha);//将位置a上的高度为ha的木块以上的木块进行归位
}
pile_dig(pa,ha,pb);//将木块a以及以上的木块堆叠到木块b上
}
for(int i=0;i<n;i++)
{
cout<<i<<":";
for(int j=0;j<s[i].size();j++)
{
cout<<" "<<s[i][j];
}
cout<<endl;
}
return 0;
}