紫书——木块问题

题意

输入n，得到编号为0~n-1的木块，分别摆放在顺序排列编号为0~n-1的位置。现对这些木块进行操作，操作分为四种。

1、move a onto b：把木块a、b上的木块放回各自的原位，再把a放到b上；

2、move a over b：把a上的木块放回各自的原位，再把a发到含b的堆上；

3、pile a onto b：把b上的木块放回各自的原位，再把a连同a上的木块移到b上；

4、pile a over b：把a连同a上木块移到含b的堆上。

分析

里面看似复杂，有四个操作，但其实抽象出来就只有两个操作，然后分别写成两个函数就差不多了。

第一个：把木块XX上面的木块归位

第二个：把木块XX及其上面的木块撂到XX堆上

因此，首先得把木块XX所在的位置（即在哪一堆，哪一高度）求出来，这时就需要再写一个函数，共三个函数

最后就是主函数上的输入输出了，（在这里吐槽一下，这题连样例都没有，如何能写输入输出呢~~~），所以输入输出看看就好（个人观点）

下面奉上代码及注释

代码

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;

const int maxn = 30;
int n;
vector<int>pile[maxn];

//找到a木块的位置，用引用方式把所在的堆跟高度传出去
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;
            }
        }
    }
}

//将木块a上方的木块归位，注意最后将a堆的高度改为h+1；
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);
}

//将a木块及其上方的木块撂到b堆上，也是记得高度问题
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) {
        cout << i << ':';
        for (int j = 0; j < pile[i].size(); ++j) {
            cout << ' ' << pile[i][j];
        }
        cout << endl;
    }
}

int main() {
    
    int a, b;
    cin >> 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;
}