【UVa101】木块问题

在学习STL，本文单纯抄代码已达到理解熟悉STL的目的

#include <cstdio>
#include <string>
#include <vector>
#include <iostream>
using namespace std;

const int maxn = 30;
int n;
vector<int> pile[maxn];  //每个pile是一个vector,可理解为声明了一个int型二维数组，第一维长度固定，第二维长度可变

//找木块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;      //p,h的值已经改变
}

//把第p堆高度为h的木块上方的所有木块移回原位
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);
}

//把第p堆高度为h及其上方的木块整体移动到p2堆的顶部
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 >> a >> s2 >> b)  //对于quit的处理，即while结束条件，暂时还不太明白
    {
        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);  //s2是over的话，不需要归位，直接移就行
        if(s1 == "move") clear_above(pa, ha);  //s1是pile的话，不需要归位，直接移就行
                                               //看着四个操作，实际就两个操作，归位和移动
        pile_onto(pa, ha, hb);
    }
    print();
    return 0;
}