UVA - 101 The Blocks Problem vector

题目大意:有n堆木块,编号为0— n-1,现在给出4种操作
mova a onto b:把a和b上面木块全部归位,再把a放到b的上面
move a over b:把a上面的木块全部归位,再把a放到b木块所在堆的顶部
pile a onto b:把b上面的木块全部归位,再把a及a上面的木块放到b上面
pile a over b:把a及上面的木块全部放到b所在木块堆的顶部

#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
#define maxn 30
vector<int> pile[maxn];
int n;
char str1[maxn], str2[maxn];

void init() {
    for(int i = 0; i <= n; i++) 
        pile[i].push_back(i);
}

void find(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 t = pile[p][i];
        pile[t].push_back(t);
    }
    pile[p].resize(h+1);
}

void clear_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, pa, pb, ha, hb;
    scanf("%d", &n);
    init();
    while(scanf("%s", str1) == 1) {
        if(strcmp(str1,"quit") == 0)
            break;
        scanf("%d%s%d", &a, str2, &b);
        find(a,pa,ha);
        find(b,pb,hb);

        if(pa == pb)
            continue;

        if(strcmp(str2,"onto") == 0)
            clear_above(pb,hb);
        if(strcmp(str1,"move") == 0)
            clear_above(pa,ha);
        clear_onto(pa,ha,pb);
    }
    print();
    return 0;
}