木块问题 the blocks problem uva 101

问题描述
思路
找到四种操作的共同点，减少代码量
归位的意思是将0号木块放在0号堆上，1号木块放于1号堆上…
代码
注：代码是紫皮书P110

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

const int maxn=100;
int n;
vector<int> pile[maxn];//pile[i]表示第i个木块堆

//找到木块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;
}

//将(p,h)上的木块归位
void clear_above(int p,int h)
{	
	//for循环进行归位
	for(int i=h+1;i<pile[p].size();i++)
	{
		int t=pile[p][i];
		pile[t].push_back(t);
	}
	//之后将p堆的大小变为h+1
	pile[p].resize(h+1);

}
//将(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);
}


//输出这n个木块堆中木块编号
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()
{
	cin>>n;
	//初始化，为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)
	{
		//可将(pa,ha)看成坐标
		int pa,pb,ha,hb;
		//获得木块a，b所在的坐标位置
		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);//将(pa,ha)及其上的块，放入堆pb
	}
	print();
	return 0;
}

运行结果
window10下

ubuntu64下运行结果（quit之后需要按crtl+D）

注：关于crtl+Z和crtl+D，参看此链接