【算法竞赛入门经典】第五章-木块问题（详细注释）

#include<cstdio>
#include<iostream>
#include<vector>
#include<string>
const int maxn = 30;
int n;//一共有n堆 
using namespace std;
vector<int>pile[maxn];//声明一个vector数组pile，一共有n堆，每一堆的高度是不确定的，又都是一个vector数组 （相当于一个二维vector数组） 

void find_block(int a,int& p,int& h)
{
	for(p=0;p<n;p++)//从第一堆开始找(pile数组的第一个元素)
	{
		for(h=0;h<pile[p].size();h++)//从第p堆的第一个元素开始查找
		{
			if(pile[p][h]==a)
			return ;//此处返回的是上文中以传引用形式传参的参数，即：返回此时的p和h 
		 } 
	} 
}

void clear_above(int a,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);//把第p堆的h+1及其上方的木块都清零（因为已经被拿走了） 
}

void pile_onto(int p1,int h,int p2)
{
	for(int i=h;i<pile[p1].size();i++)
	{
		pile[p2].push_back(pile[p1][i]);
	}
	pile[p1].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 = 0;
	cin>>n;
	string s1,s2;
	while(cin>>s1>>a>>s2>>b)//输入对应操作即元素 
	{
		int pa,ha,pb,hb = 0;
		find_block(a,pa,ha);//找到a木块所在的堆pa，和在pa堆中的高度ha 
		find_block(b,pb,hb);
		if(pa==pb) continue;
		if(s1=="onto") clear_above(a,pa,hb);//clear_above函数实现把第p堆高度为h的木块上方的所有木块移回原位 
		if(s2=="move") clear_above(b,pb,hb);
		pile_onto(pa,ha,pb);//pile_onto函数实现把第p堆高度为h及其上方的木块整体移动到p2堆的顶部 
	}
	print();//自己编写的打印vector数组的函数
	return 0; 
}