E - The Blocks Problem
从左到右有n个木块,编号为0到n-1,要求模拟一下4种操作(a,b均为木块编号)。
move a noto b:把a和b上方的木块全部归位,然后把a放在b上面。
move a over b:把a上方的木块全部归位,然后把a放在b所在木块堆的顶部。
pile a onto b:把b上方的木块全部归位,然后把a及上面的木块整体放在b上面。
pile a over b:把a及上面的木块整体放在b所在木块堆的顶部。
qiut结束。
从底到顶输出木块的排列顺序。
题解:用vector数组来做,a.sizeof()代表大小,a.resize()改变大小,a.push_back()向尾部添加元素,a.pop_back删除最后一个元素。
#include <cstdio>
#include <string>
#include <vector>
#include <iostream>
using namespace std;
const int maxn = 30;
int n;
vector<int>pile[maxn];
void find_block(int a,int &p,int &h)
{
for(p = 0;p < n;p++)
{
for(h = 0;h < pile[p].size();h++)
{
if(pile[p][h] == a)
{
return;
}
}
}
}
void clear_above(int p,int h)
{
for(int i = h + 1;i < pile[p].size();i++)
{
int b = pile[p][i];
pile[b].push_back(b);
}
pile[p].resize(h + 1);
}
void pile_onto(int p,int h,int p2)
{
for(int i = h;i < pile[p].size();i++)
{
pile[p2].push_back(pile[p][i]);
}
pile[p].resize(h);
}
void print()
{
for(int i = 0;i < n;i++)
{
printf("%d:", i);
for(int j = 0;j < pile[i].size();j++)
{
printf(" %d",pile[i][j]);
}
printf("\n");
}
}
int main()
{
int a,b;
scanf("%d",&n);
string s1,s2;
for(int i = 0;i < n;i++) pile[i].push_back(i);
while(cin >> s1 >> a >> s2 >> b)
{
int pa,pb,ha,hb;
find_block(a,pa,ha);
find_block(b,pb,hb);
if(pa == pb) continue;
if(s2 == "onto") clear_above(pb,hb);
if(s1 == "move") clear_above(pa,ha);
pile_onto(pa,ha,pb);
}
print();
return 0;
}