UVa 101 The Blocks Problem

UVa 101 The Blocks Problem

#include<cstdio>
#include<cstring>
#include<iostream>
#include<stack>
using namespace std;

const int maxn = 26;
stack<int> q[maxn];
int pos[maxn]; //记录每个方块的位置
int chg[maxn]; //用于临时存放从栈中pop出的数

void clear(int i){ //将目标方块上的方块放回原位
int p = pos[i];
int temp;
while(q[p].top()!=i){
temp = q[p].top();
pos[temp] = temp;
q[p].pop();
q[temp].push(temp);
}
}

void pile(int a, int b){ //将a及以上的方块堆放在b所在的方块堆上
int cnt = 0;
int temp;
int p1 = pos[a];
int p2 = pos[b];
while(q[p1].top()!=a){
temp = q[p1].top();
q[p1].pop();
pos[temp] = p2;
chg[cnt++] = temp;
}
q[p1].pop();
pos[a] = p2;
q[p2].push(a);
while(--cnt>=0)
q[p2].push(chg[cnt]);
}

int main(){
int n,a,b;
char op1[5],op2[5];
scanf("%d",&n);
for(int i = 0;i<n;i++){
q[i].push(i);
pos[i] = i;
}
while(scanf("%s",op1)&&op1[0]!='q'){
scanf("%d %s %d",&a,op2,&b);
if(a==b||pos[a]==pos[b])
continue;
if(op1[0]=='m') //move情况下清空a上的方块
clear(a);
if(op2[1]=='n') //onto情况下清空b上的方块
clear(b);
pile(a,b);
}
int cnt;
for(int i = 0;i<n;i++){
printf("%d:",i);
cnt = 0;
while(!q[i].empty()){
chg[cnt++] = q[i].top();
q[i].pop();
}
for(int j = cnt-1;j>=0;j--)
printf(" %d",chg[j]);
printf("\n");
}
}