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：

、resize(n)

调整容器的长度大小，使其能容纳n个元素。

如果n小于容器的当前的size，则删除多出来的元素。

否则，添加采用值初始化的元素。

、 resize(n, t)

多一个参数t，将所有新添加的元素初始化为t。

code：

 #include <iostream>

 #include <string>

 #include <vector>

 using namespace std;

 const int MAXN = ;

 vector<int> pile[MAXN];

 int n;

 void findBlock(int a, int& p, int& h)

 {

     for (p = ; p < n; ++p)

     {

         for (h = ; h < pile[p].size(); ++h)

         {

             if (pile[p][h] == a)

                 return;

         }

     }

 }

 void clearAbove(int p, int h)

 {

     for (int i = h + ; i < pile[p].size(); ++i)

     {

         int b = pile[p][i];

         pile[b].push_back(b);

     }

     pile[p].resize(h + );

 }

 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 = ; 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 = ; i < n; ++i)

     {

         cout << i << ":";

         for (int j = ; j < pile[i].size(); ++j)

             cout << " " << pile[i][j];

         cout << endl;

     }

     return ;

 }