E - The Blocks Problem

E - The Blocks Problem

从左到右有n个木块，编号为0到n-1，要求模拟一下4种操作（a,b均为木块编号）。

move a noto 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所在木块堆的顶部。

qiut结束。

从底到顶输出木块的排列顺序。

题解：用vector数组来做，a.sizeof()代表大小，a.resize()改变大小，a.push_back()向尾部添加元素，a.pop_back删除最后一个元素。

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

const int maxn = 30;

int n;

vector<int>pile[maxn];

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 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);
}

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;
    scanf("%d",&n);
    string s1,s2;
    for(int i = 0;i < n;i++) pile[i].push_back(i);
    while(cin >> s1 >> 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;
}