紫书STL初步-木块问题

题目链接
题目大意:有0-n-1块木块,要求模拟4种操作
mova 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以及上面木块放置b所在木块顶部
程序设计分析:
对于 mova a onto b 我们需要完成的
1 找到a
2 将a上方木块复位
3 找到b
4 将b上方木块复位
5 将a放到b上
move a over b
1 找到a
2 将a上方木块归位
3 找到b
4 直接放置顶部
pile a onto b
1 找到b
2 将b顶上木块复位
3 找到a
4将a及以上木块直接放置b上
pile a over b
1找到a
2找到b
3将a及以上直接放置b顶上
发现操作有大量重复内容根据需求可以分析出需要以下几个函数
1 找到木块x所在的木块堆
2 将木块x以上的木块复位
3将木块x及以上的木块整体移动至顶部.
对于每一种操作我们都需要通过1 知道a,b所在的位置
如果是move 我们需要通过2复位a上木块
如果是onto我们需要通过 2复位b上木块
最后通过三将a放置b

#include<iostream>
#include<algorithm>
#include<string>
#include<vector>
using namespace std;
vector<int>pile[30];
int n;
void clere(int p,int h)
{
    for(int i=h+1;i<pile[p].size();i++)
    {
        int b=pile[p][i];
        pile[b].push_back(b);
    }
    pile[p].resize(h+1);
}
void find(int x,int& p,int& h)
{
    for( p=0;p<n;p++)
       for(h=0;h<pile[p].size();h++)
       {
           if(pile[p][h]==x)return ;
       }
}
void move_(int pa,int ha,int pb)
{
    for(int i=ha;i<pile[pa].size();i++)
    {
        int b=pile[pa][i];
        pile[pb].push_back(b);
    }
    pile[pa].resize(ha);
}
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");
    }
}
int main()
{   
    
    cin>>n;
    for(int i=0;i<n;i++)
        pile[i].push_back(i);
    string s1,s2;
    int a,b;
    while(cin>>s1>>a>>s2>>b)
    {   if(s1=="quit")break;
        
        int pa,ha,pb,hb;
        find(a,pa,ha);
        find(b,pb,hb);
        if(pa==pb)continue;
        if(s1=="move")clere(pa,ha);
        if(s2=="onto")clere(pb,hb);
        
        move_(pa,ha,pb);
    }
    print();
    return 0;
}