简单说下规则:
1.move a onto b
a和b都是积木的编号,先将a和b上面所有的积木都放回原处,再将a放在b上。
2.move a over b
a和b都是积木的编号,先将a上面所有的积木放回原处,再将a放在b上。(b上原有积木不动)
3.pile a onto b
a和b都是积木的编号,将a和其上面所有的木块组成的一摞整体移动到b上。在移动前要先将b上面所有的木块都放回原处。移动的一摞积木要保持原来的顺序不变。
4.pile a over b
a和b都是积木的编号,将a和其上面所有的积木组成的一摞整体移动到b所在一摞积木的最上面一个积木上。移动的一摞积木要保持原来的顺序不变。
仔细观察上面的规则存在着一些共同点:
1.如果是onto的话就是要把b上面的木块全部都放回原处
2.pile的指令和move的指令都是要把a(或者a和a以上)的积木放到b上
3.move,是移动a上面的积木,而不是b上面的积木,
由此,就可以根据共同点来精简代码。
代码如下:
#include <iostream>
#include <string>
#include <vector>
using namespace std;
const int maxn = 30;
vector<int> pile[maxn];
int n, a, b;
string s1, s2;
void find_block(int a, int &p, int &h) { //找出木块p所在的pile和height,以引用的方式返回调用者
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) { //把第p堆高度为h的木块上方的所有木块移回原地
for (int i = h + 1; i < pile[p].size(); i++) {
int b = pile[p][i];
pile[b].push_back(b); //把木块b放回原位
}
pile[p].resize(h+1); //pile只应保留下标0~h的元素
}
void pile_onto(int p, int h, int p2) { //把第p堆高度为h及其上方的木块整体移动到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++) {
cout << i << ":";
for (int j = 0; j < pile[i].size(); j++)
cout << " " << pile[i][j];
cout << endl;
}
}
int main() {
cin >> n;
pile
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;
}