Problem B The Blocks Problem（vector的使用）

题目链接：Problem B

题意：有n块木块，编号为0~n-1，要求模拟以下4种操作（下面的a和b都是木块编号）

1. move a onto b: 把a和b上方的木块全部归位，然后把a摞在b上面。

2. move a over b: 把a上方的木块全部归位，然后把a放在b所在木块堆的顶部。

3. pile a onto b: 把b上方的木块全部归位，然后把a及上面的木块整体摞在b上面。

4. pile a over b: 把a及上面的木块整体摞在b所在木块堆的顶部。

遇到quit时终止一组数据，忽略非法指令。

思路：每个木块堆都用一个vector来维护，move操作都要对a归位，onto操作都要对b归位可以一起处理，四种操作最后都是把a及以上的摞在b的顶上，只不过前两种a上面的为空。

note：
1、resize(n) 
调整容器的长度大小，使其能容纳n个元素。
如果n小于容器的当前的size，则删除多出来的元素。
否则，添加采用值初始化的元素。
2、 resize(n, t)
多一个参数t，将所有新添加的元素初始化为t。

code：

#include <iostream>
#include <string>
#include <vector>
using namespace std;
const int MAXN = 30;
vector<int> pile[MAXN];
int n;

void findBlock(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 clearAbove(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);
}

void pileOnto(int pa, int ha, int pb)
{
    for (int i = ha; i < pile[pa].size(); ++i)
        pile[pb].push_back(pile[pa][i]);
    pile[pa].resize(ha);
}

int main()
{
    cin >> n;
    for (int i = 0; i < n; ++i) pile[i].push_back(i);
    int a, b;
    string s1, s2;
    while (cin >> s1)
    {
        if (s1 == "quit") break;
        cin >> a >> s2 >> b;
        int pa, pb, ha, hb;
        findBlock(a, pa, ha);
        findBlock(b, pb, hb);
        if (pa == pb) continue;
        if ("move" == s1) clearAbove(pa, ha);
        if ("onto" == s2) clearAbove(pb, hb);
        pileOnto(pa, ha, pb);
    }
    for (int i = 0; i < n; ++i)
    {
        cout << i << ":";
        for (int j = 0; j < pile[i].size(); ++j)
            cout << " " << pile[i][j];
        cout << endl;
    }
    return 0;
}