从左到右有n个木块(编号为0~n-1),要求模拟以下4种操作(下面的a和b都是木块编号)。
move a onto 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所在堆的顶部
用quit命令退出并打印出最后木块堆的状态。 题目分析:每个木块堆的高度不确定,适合用vector来保存。而木块堆的个数不超过n,所以用一个数组来存就可以了。
代码如下:
#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; cin >> n; string s1, s2; for(int i = 0; i < n; i++) pile[i].push_back(i); while(cin >> s1) { if(s1 == "quit") break; cin >> 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; }