101 The Blocks Problem （vector二维）

#include <iostream>
#include <cstdio>
#include <vector>
#include <string>
using namespace std;
int n;
vector<int>pile[30];//重点在于记录实时位置，vector这种可以动态申请的数组正好可以当作二维数组用。第一维堆第二维层

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;//void中的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);   //把标号为t的木块放回原位，即t堆

  }
 pile[p].resize(h + 1);    //重新定义p堆的长度

}
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;
       string s1, s2;
       scanf("%d", &n);
       for (int i = 0; i<n; i++) pile[i].push_back(i);
       while (cin >> s1) {//通过对题意的归纳可以通过函数重用减少多余代码
             if (s1 == "quit") break;
             cin >> a >> s2 >> b;
             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);
             if (s1 == "move") clear_above(pa, ha);
             pile_onto(pa, ha, pb);

  }
       print();
       return 0;

}