The Blocks Problem （vector使用）

 

 题意：给定n个木块，根据操作指令进行操作。

move a onto b:把a和b木块上边的所有木块移到初始位置，然后把a移到b木块上

move a over b:把a木块上方所有木块移动到初始位置，把a放到b上方

pile a onto b:，把b木块上方所有木块移到初始位置，把a木块上方所有木块连同a移动到b上方，并且a保持原序列

pile a over b:把a木块上方所有木块连同a移到b上方，并且保持原序列不变

#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue> 
#include<vector>
using namespace std;
const int N = 100;
int a[N];
int n;

vector<int> v[N];

void find(int x,int& p,int& h)//找到该木块所在位置
{
	for(p=0;p<n;p++)//注意这里的p和h不能加int
	{
		for(h=0;h<v[p].size();h++)
		{
			if(v[p][h]==x) return;
		}
	}
}
void initial(int p,int h)//把所有木块上方的木块移动到初始位置
{
	for(int i=h+1;i<v[p].size();i++)
	{
		int b=v[p][i];
		v[b].push_back(b);
	}
	v[p].resize(h+1);//改变大小，只取0~h的数
}

void file(int p,int h,int p2)//整体移动木块
{
	for(int i=h;i<v[p].size();i++)
	{
		int b=v[p][i];
		v[p2].push_back(b);
	}
	v[p].resize(h);//改变大小，只取0~h-1的数
}
int main()
{
    cin>>n;
    for(int i=0;i<n;i++) v[i].push_back(i);
    
    string s1,s2;
    int a,b;
    while(1)
    {
    	cin>>s1;
    	if(s1=="quit") break;
    	cin>>a>>s2>>b;
    	int ha,hb,pa,pb;
    	
    	find(a,pa,ha);
    	find(b,pb,hb);
    	
    	if(a==b || pa==pb) continue;
    	
    	if(s2 == "onto") initial(pb,hb);
    	
    	if(s1 == "move") initial(pa,ha);
    	
    	
    	file(pa,ha,pb);
	}
	
	for(int i=0;i<n;i++)
	{
		printf("%d:",i);
		for(int j=0;j<v[i].size();j++)
		{
			printf(" %d",v[i][j]);
		}
		printf("\n");
	}
	return 0;


}