UVA101 -- The Blocks Problem

簡單說下規則：
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;
}