木块问题（UVa101）

 

　　题目具体描述见：https://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=37

将操作分解为基本步骤，编写函数，重复调用

C++11代码如下：

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

const int maxn =25;
vector<int> pile[maxn];
int n;
void find_block(int a, int& p, int& h) {  //找到木块所在的位置和高度（从0开始）
    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) {  //将p位置高度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);  //保留 0-h高度的木块
}

void pile_onto(int p, int p2,int h) {   //将p位置处从h高度的木块全部移动到p2顶部
    for (int i = h; i < pile[p].size(); i++) {
        pile[p2].push_back(pile[p][i]);
    }
    pile[p].resize(h);  //保留高度0~h-1的木块
}

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 ha, pa, hb, pb;
    cin >> n;
    for (int i = 0; i < n; i++) pile[i].push_back(i);
    string s1, s2;
    int a, b;
    while (cin >> s1 >> a >> s2 >> b) {
        find_block(a, pa, ha);
        find_block(b, pb, hb);
        if (pa == pb) continue;
        if (s1 == "move") clear_above(pa, ha);
        if (s2 == "onto") clear_above(pb, hb);
        pile_onto(pa, pb, ha);
    }
    print();
    return 0;
}