POJ 1208 The Blocks Problem (UVA 101)

题意：n堆木块，每堆一个，编号0到n-1，现在给你一些操作指令，分别是：

move a onto b：将a,b之上的木块复位，再将a放到b上；

move a over b：将a之上的木块复位，再将a放到包含b的堆的堆顶；

pile a onto b：将b之上的木块复位，再将a及a之上的木块放到b上；

pile a over b：将a及a以上的木块放到包含b的堆的堆顶。

输出最后的状态即可。

解题思路：用数组模拟即可，要注意的的是：当a=b是，不用操作；当a与b在同一堆中，该操作属于错误指令，直接忽略。还有，可以证明，将某个木块上的其他木块复位时，不用担心原来的位置被其他木块占了。

Memory: 692K		Time: 0MS

#include <iostream>
#include <stdio.h>
#include <string.h>

using namespace std;
int s[30][30],sn[30],n,a,b;//堆，堆中木块数，总堆数
char op[5],dr[5];

void init()//初始化
{
for(int i=0;i<n;i++)
{
sn[i]=1;
s[i][1]=i;
}
}

int find(int x)//找到标号为x的木块所在的堆
{
for(int i=0;i<n;i++)
for(int j=1;j<=sn[i];j++)
{
if(s[i][j]==x) return i;
}
}

void reset(int x)//将木块x之上的木块复位
{
int i=find(x);
for(int j=1;j<=sn[i];j++)
{
if(s[i][j]==x)
{
int p=j+1;
for(;p<=sn[i];p++)
{
int y=s[i][p];
s[y][1]=y;
sn[y]=1;
}
sn[i]=j;
break;
}
}
}

void print()//输出所有堆状态
{
for(int i=0;i<n;i++)
{
printf("%d:",i);
for(int j=1;j<=sn[i];j++)
{
printf(" %d",s[i][j]);
}
printf("\n");
}
}

int main()
{
//    freopen("in.txt","r",stdin);
while(scanf("%d",&n)!=EOF)
{
int x,y;
init();
while(scanf("%s",op))
{
if(!strcmp(op,"quit")) break;
scanf("%d%s%d",&a,dr,&b);
if(a==b) continue;
if(!strcmp(op,"move") && !strcmp(dr,"onto"))
{
x=find(a);
y=find(b);
if(x==y) continue;//如果a,b在同一堆，则操作忽略，下同
reset(a);
reset(b);
sn[y]++;
s[y][sn[y]]=a;
sn[x]--;
}
else if(!strcmp(op,"move") && !strcmp(dr,"over"))
{
x=find(a);
y=find(b);
if(x==y) continue;
reset(a);
sn[y]++;
s[y][sn[y]]=a;
sn[x]--;
}
else if(!strcmp(op,"pile") && !strcmp(dr,"onto"))
{
y=find(b);
x=find(a);
if(x==y) continue;
reset(b);
for(int i=1;i<=sn[x];i++)
{
if(s[x][i]==a)
{
int tmp=sn[x];
sn[x]=i-1;
for(int j=i;j<=tmp;j++)
{
sn[y]++;
s[y][sn[y]]=s[x][j];
}
break;
}
}
}
else if(!strcmp(op,"pile") && !strcmp(dr,"over"))
{
x=find(a);
y=find(b);
if(x==y) continue;
for(int i=1;i<=sn[x];i++)
{
if(s[x][i]==a)
{
int tmp=sn[x];
sn[x]=i-1;
for(int j=i;j<=tmp;j++)
{
sn[y]++;
s[y][sn[y]]=s[x][j];
}
break;
}
}
}
}
print();
}
return 0;
}