The Blocks Problem UVA - 101

题目链接：https://vjudge.net/problem/UVA-101#author=Zsc1615925460

题目大意：从左到右右n个木块，编号为0-n-1，要求模拟一下四种操作。

• move a onto b: 把a和b上方的木块全部归位，然后把a摞在b上面。
• move a over b: 把a上方的木块全部归位，然后把a放在b所在的木块堆的顶端
• pile a onto b: 把b上方的木块全部归位，然后把a以及以上的所有木块摞在b上面。
• pile a over b：把a以及以上的木块摞在b所在木块堆的顶端。
• quit：退出。
  a和b在同一堆的指令为非法指令，应忽略。

分析：要选好方法，用容器很简单，要注意控制容器的大小，resize()。

#include <cstdio>
#include <string>
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
const int maxn = 50;
int n;
vector<int> arr[maxn];

void Search(int a, int &l, int &h) {
    for(l = 0; l < n; l++) {
        for(h = 0; h < arr[l].size(); h++) {
            if(arr[l][h]==a) return;
        }
    }
}

void Return(int l1, int h1) {
    for(int i = h1+1; i < arr[l1].size(); i++)
        arr[arr[l1][i]].push_back(arr[l1][i]);
    arr[l1].resize(h1+1);
}
void Pover(int l1, int h1, int l2) {
    for(int i = h1; i < arr[l1].size(); i++)
        arr[l2].push_back(arr[l1][i]);
    arr[l1].resize(h1);
}

int main()
{
    freopen("i.txt","r",stdin);
    int a,b;
    cin >> n;
    for(int i = 0; i < n; i++)
        arr[i].push_back(i);
    string s1,s2;
    while(cin >> s1 >> a >> s2 >> b && s1!="quit") {
        int l1,l2,h1,h2;
        Search(a,l1,h1);
        Search(b,l2,h2);
        if(l1==l2) continue;
        if(s1=="move" && s2=="onto")
            Return(l1,h1),Return(l2,h2);
        else if(s1=="move" && s2=="over")
            Return(l1,h1);
        else if(s1=="pile" && s2=="onto")
            Return(l2,h2);
        Pover(l1,h1,l2);
    }
    for(int i = 0; i < n; i++) {
        cout << i << ":";
        for(int j = 0; j < arr[i].size(); j++)
             cout << " " << arr[i][j];
        if(i<n)cout << endl;
    }
    return 0;
}