uva101 the blocks pronlem

对不定长数组的操作：

push_back()在尾部添加元素

pop_back()在尾部删除元素

clear()清空

resize()改变大小

empty()测试是否为空

#include<cstdio>
#include<string>
#include<vector>
#include<iostream>
using namespace std;
const int maxn = 30;
int n;
vector<int> pile[maxn];//每个pile[i]是一个不定长的数组
//找木块a所在的pile和height,以引用的形式返回调用者
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;//找到a后，引用的形式返回它的pile和height
}
//把第p堆高度为h的木块上方的所有木块移回原位
void clear_above(int p, int h) {
    for(int i = h+1; i < pile[p].size(); i++) {//从要移元素的height+1开始移
        int b = pile[p][i];//把要移的值暂时保留在变量b中
        pile[b].push_back(b);//进行归位
    }
    pile[p].resize(h+1);//改变pile[p]的大小
}
//把第p堆高度为h及其上方的木块整体移动到p2堆得顶部
void pile_onto (int p, int h, int p2) {
    for(int i = h; i < pile[p].size(); i++)//从第h个开始移p
        pile[p2].push_back(pile[p][i]);//每个p用ush_back()方法进行添加移入p2
    pile[p].resize(h);//最后别忘了改变移动后的p的大小
}
//c此部是输出
void print() {
    for(int i = 0; i < n; i++) {//n个木块的位置
        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 >> a >> s2 >> b) {//读入每一组操作
        int pa, pb, ha, hb;
        find_block(a, pa, ha);
        find_block(b, pb, hb);//找到a,p所在的pilr和height
        if(pa == pb) continue;//这是非法输入
        //在所给的操作中抽象出两个一样的操作
        if(s2 == "onto") clear_above(pb, hb);//一是"onto"时，归位b上的全部木块
        if(s1 == "move") clear_above(pa, ha);//二是"move"时，归位a上的全部木块
        pile_onto(pa, ha, pb);//所有的都是把a移动到b上
    }
    print();
    return 0;
}