101 - The Blocks Problem

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

const int Max=30;
struct node
{
	int stack[Max];
	int top;//+1
	node():top(0){}
}s[Max];

int n;

void init()
{
	for(int i=0;i<n;i++)
		s[i].stack[s[i].top++]=i;
}
int getCommand(string &s1,string &s2)
{
	if(s1=="move" && s2=="onto")
		return 1;
	else if(s1=="move" && s2=="over")
		return 2;
	else if(s1=="pile" && s2=="onto")
		return 3;
	else if(s1=="pile" && s2=="over")
		return 4;
	return 0;
}
int getPos(int x,int &pos,int &lev)
{
	for(int i=0;i<n;i++)
	{
		for(int j=0;j<s[i].top;j++)
			if(s[i].stack[j]==x)
			{
				pos=i;
				lev=j;
				return 1;
			}
	}
	return 0;
}
void handle(string &s1,int a,string &s2,int b)
{
	int sort=getCommand(s1,s2);
	int a_pos,b_pos;
	int a_lev,b_lev;
	int i;
	int temp;
	getPos(a,a_pos,a_lev);
	getPos(b,b_pos,b_lev);
	if(a_pos==b_pos)
		return ;
	switch(sort)
	{
	case 1:
		for(i=a_lev+1;i<s[a_pos].top;i++)
		{
			temp=s[a_pos].stack[i];
			s[temp].stack[s[temp].top++]=temp;
		}
		s[a_pos].top=a_lev+1;
		for(i=b_lev+1;i<s[b_pos].top;i++)
		{
			temp=s[b_pos].stack[i];
			s[temp].stack[s[temp].top++]=temp;
		}
		s[b_pos].top=b_lev+1;
		s[b_pos].stack[s[b_pos].top++]=s[a_pos].stack[--s[a_pos].top];
		break;
	case 2:
		for(i=a_lev+1;i<s[a_pos].top;i++)
		{
			temp=s[a_pos].stack[i];
			s[temp].stack[s[temp].top++]=temp;
		}
		s[a_pos].top=a_lev+1;
		s[b_pos].stack[s[b_pos].top++]=s[a_pos].stack[--s[a_pos].top];
		break;
	case 3:
		for(i=b_lev+1;i<s[b_pos].top;i++)
		{
			temp=s[b_pos].stack[i];
			s[temp].stack[s[temp].top++]=temp;
		}
		s[b_pos].top=b_lev+1;
		for(i=a_lev;i<s[a_pos].top;i++)
		{
			temp=s[a_pos].stack[i];
			s[b_pos].stack[s[b_pos].top++]=temp;
		}
		s[a_pos].top=a_lev;
		break;
	case 4:
		for(i=a_lev;i<s[a_pos].top;i++)
		{
			temp=s[a_pos].stack[i];
			s[b_pos].stack[s[b_pos].top++]=temp;
		}
		s[a_pos].top=a_lev;
		break;
	default:
		break;
	}
}

int main()
{
	/*
	freopen("101.in","r",stdin);
	freopen("101.out","w",stdout);
	//*/
	scanf("%d",&n);
	getchar();
	init();
	string command;
	while(getline(cin,command) && command!="quit")
	{
		int a,b;
		string s1,s2;
		istringstream in(command);
		in>>s1;
		in>>a;
		in>>s2;
		in>>b;
		handle(s1,a,s2,b);
	}
	for(int i=0;i<n;i++)
	{
		printf("%d:",i);
		for(int u=0;u<s[i].top;u++)
			printf(" %d",s[i].stack[u]);
		//if(s[i].top>0)
			//printf("%d",s[i].stack[s[i].top-1]);
		printf("\n");
	}
	return 0;
}