木块问题（The Blocks Problem. UVa 101）

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

• move a onto b: 把a和b上方的木块全部归位，然后把a摞到b上面。
• move a over b: 把a上方的木块全部归位，然后把a及上面的木块整体摞到b上面。
• pile a onto b: 把b上方的木块全部归位，然后把a及a上面的木块摞在b上面。
• pile a over b: 把a及上面的木块整体摞在b所在木块堆的顶部。

遇到quit 时终止一组数据。a和b在同一堆的指令是非法指令，应当忽略。

所有操作结束后，输出每个位置的木块列表，按照从底部到顶部的顺序排列。

 

分析：

每个木块堆的高度不确定，所以用vector来保存很合适；而木块堆的个数不超过n,所以用一个数组来存就可以了。

然后，可以把4个操作总结为：

当出现move时，会把a全部归位；当出现onto时，会把b全部归位。

代码如下：

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

const int maxn = 30;
int n;
vector<int> pile[maxn]; //每个pile[i]是一个vector

//找木块a所在的pile和height,以引用的形式返回调用者
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;
            }
        }
    }
}

//把第p堆高度为h的木块上方的所有木块移回原位
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); //pile只应保留下标0-h的元素 
}

//把第p堆高度为h及其上方的木块整体移动到p2堆的顶部
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;
    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 == "over") clear_above(pa,ha);
        pile_onto(pa,ha,pb);
    }
    print();
    return 0;
}

 

也可以选择调试模式，来观察下，代码如下：

    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){
        if(s1 == "quit" || s2 == "quit"){
            break;
        }
        int pa,pb,ha,hb;
        find_block(a,pa,ha);
        printf("pa:%d,ha:%d \n",pa,ha);
        find_block(b,pb,hb);
        printf("pb:%d,hb:%d \n",pb,hb);
        if(pa == pb) continue;
        if(s2 == "onto"){
            clear_above(pb,hb);
            printf("onto: \n");
            print();
        }
        if(s1 == "move"){
            clear_above(pa,ha);
            printf("move: \n");
            print();
        }
        pile_onto(pa,ha,pb);
        printf("pile onto: \n");
        print();
    }
    print();
    return 0;

 

 

 题目地址