#include<iostream>
#include<vector>
using namespace std;
vector<int> pile[30];
vector<int> tem;
int n;
void print()
{
for(int i=0;i<n;i++)
{
printf("%d:",i);
for(int j=0;j<pile[i].size();j++) {
printf(" %d",pile[i][j]);
}
printf("\n");
}
}
void find(int a,int &posx,int &posy)
{
for(int i=0;i<n;i++){
int x=pile[i].size() ;
for(int j=0;j<x;j++)
{
if(pile[i][j]==a)
{
posx=i;posy=j;
return;
}
}
}
}
void clear_above(int x,int y)
{
for(int j=pile[x].size()-1;j>y;j--)
{
pile[pile[x][j]].push_back(pile[x][j]);
pile[x].pop_back();
}
}
void pile_(int ax,int ay)
{
int x=pile[ax].size();
for(int j=x-1;j>=ay;j--)
{
tem.push_back(pile[ax][j] );
pile[ax].pop_back();
}
}
int main()
{
int blo;
string s1,s2;
int a,b;
int ax,ay,bx,by;
while(cin>>n)
{
for(int i=0;i<30;i++)
{
pile[i].clear() ;
}
tem.clear();
for(int i=0;i<n;i++)
{
pile[i].push_back(i);
}
while(cin>>s1)
{
if(s1=="quit") break;
cin>>a>>s2>>b;
find(a,ax,ay);
find(b,bx,by);
if(ax==bx) continue;//如果注释掉这句话就会wa。因为当要处理的两个块在同一列上时,所有的操作都无效= =
if(s1=="move") clear_above(ax,ay);
if(s2=="onto") clear_above(bx,by);
pile_(ax,ay);
while(tem.empty()==0)
{
blo=tem.back();
pile[bx].push_back(blo);
tem.pop_back();
}
}
print();
}
return 0;
}
这道题用到了引用传参,在这里很方便。本来是用全局变量来写,但是引用传参更方便。
这道题四种指令可以简化来看,move跟onto需要清空,之后统一用pile,哪怕上面没有方块。这里题意有点不清楚,题目说还原上面的方块,是还原上次移动的位置还是最开先的位置,没有做交代。然而实际上是还原最开先的位置,坑了我一下。
同时也用到了新的数据类型vector数组,在思路上比vector < vector< > > 更好理解,虽然他们两个表示的模型基本相同,但是我更喜欢vector数组。
然后参考了别人的代码,有用到vector的resize,也可以简化代码。参考下面的博客
http://blog.youkuaiyun.com/ubunfans/article/details/8447833