其实四条命令可以简化为两个操作:
1. 把某个块上面的块放回原位;
2. 把某个块及其上的块放到另一个块所在的堆顶。
还可以加一个数组保存每个块的所在的堆,这样写出来的代码比书上还间接哈哈哈。
第一次WA是多输出了换行,第二次是以为有多组数据,代码如下:
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn = 30;
int n,pos[maxn];
char cmd[20];
vector<int>bk[maxn];
void ret(int a,int pa){
for(int i = bk[pa].size() - 1;bk[pa][i] != a;i--){
int t = bk[pa][i];
bk[t].push_back(t);
pos[t] = t;
bk[pa].pop_back();
}
}
void put(int a,int pa,int pb){
int i = 0;
while(bk[pa][i] != a) i++;
for(int j = i;j < bk[pa].size();j++){
int t = bk[pa][j];
bk[pb].push_back(t);
pos[t] = pb;
}
for(int j = bk[pa].size() - 1;j >= i;j--) bk[pa].pop_back();
}
int main(){
scanf("%d",&n);
for(int i = 0;i < n;i++){
bk[i].push_back(i);
pos[i] = i;
}
while(fgets(cmd,15,stdin) && cmd[0] != 'q'){
int a = cmd[5] - '0',b = cmd[12] - '0';
if(a == b || pos[a] == pos[b]) continue;
if(cmd[0] == 'm') ret(a,pos[a]);
if(cmd[8] == 'n') ret(b,pos[b]);
put(a,pos[a],pos[b]);
}
for(int i = 0;i < n;i++){
printf("%d:",i);
for(int j = 0;j < bk[i].size();j++) printf(" %d",bk[i][j]);
putchar('\n');
}
return 0;
}
然而还是WA了,最后发现是读取数据的问题,我是把一条命令全部读到一个字符串里面,还觉得我这比他们的更简洁哈哈哈,没想到两位数。。。下面是AC代码:
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn = 30;
int n,a,b,pos[maxn];
char cmd1[5],cmd2[5];
vector<int>bk[maxn];
void ret(int a,int pa){
for(int i = bk[pa].size() - 1;bk[pa][i] != a;i--){
int t = bk[pa][i];
bk[t].push_back(t);
pos[t] = t;
bk[pa].pop_back();
}
}
void put(int a,int pa,int pb){
int i = 0;
while(bk[pa][i] != a) i++;
for(int j = i;j < bk[pa].size();j++){
int t = bk[pa][j];
bk[pb].push_back(t);
pos[t] = pb;
}
for(int j = bk[pa].size() - 1;j >= i;j--) bk[pa].pop_back();
}
int main(){
scanf("%d",&n);
for(int i = 0;i < n;i++){
bk[i].push_back(i);
pos[i] = i;
}
while(scanf("%s%d%s%d",cmd1,&a,cmd2,&b) == 4){
if(a == b || pos[a] == pos[b]) continue;
if(cmd1[0] == 'm') ret(a,pos[a]);
if(cmd2[1] == 'n') ret(b,pos[b]);
put(a,pos[a],pos[b]);
}
for(int i = 0;i < n;i++){
printf("%d:",i);
for(int j = 0;j < bk[i].size();j++) printf(" %d",bk[i][j]);
putchar('\n');
}
return 0;
}