例题 5-2 木块问题 （The Blocks Problem） UVa 101

题目：给你n个方块，有四种操作：

            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所在堆的上面。

仔细分析一下就会发现 其实只有两种操作 一个是 清空 a或b或(a和b) 上面的木块,二是 pile[a] to pile[b](单独的a或者b也可以看做pile)；

实现：

#include<iostream>
#include<vector>
#include<algorithm>
#include<string>
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)//把a/b上方的木块全部归位  注意：这里的h从0开始 所以 当前位置的元素个数为最大h+1
	{
		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)//把 a及a以上 整体移到b上（b上面是否归位取决于指令）
	{
		for(int i = h; i < pile[p].size(); i++ ){
			pile[p2].push_back(pile[p][i]);
		}	
		pile[p].resize(h);//修改此时的a的大小
	}	
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()
	{
	//	freopen("C:\\Users\\zhangwei\\Desktop\\input.txt","r",stdin);
		cin >> n;	
		string s1, s2;
		int a, b;
		for(int i = 0; i < n; i++ ) pile[i].push_back(i);
		while(cin >> s1 >> a >> s2 >> b){
			if(s1 == "quit")
				break;
			int pa, pb, ha, hb;
			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, ha, pb);
		}
		print();
		return 0;
	}