算法之路二：刘汝佳算法竞赛入门经典：STL不定长数组 木块问题 UVA101

#include<cstdio>
#include<string>
#include<vector>
#include<iostream> 
using namespace std;

const int maxn=30;
int n;
vector<int> pile[maxn];//vector是模板 要用<>声明 
                        //最多maxn堆木块，每堆个数不确定 

void find_block(int a,int &p,int &h)//找到木块a所在的堆p和高度h ，以引用形式返回 
{
    for(p=0;p<n;p++)        //寻找所在堆 
        for(h=0;h<pile[p].size();h++) //寻找所在高度 
            if(pile[p][h]==a) return;
}

void clear_above(int p,int h) // 把第p堆高度为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);//大小改变为h+1 
 } 

 void pile_onto(int p,int h,int p2)//把第p堆高度为h及其上方的木块整体移动到p2堆的顶部 
 {
    for(int i=h;i<pile[p].size();i++)
    pile[p2].push_back(pile[p][i]);//一个一个放到p2顶部 
    pile[p].resize(h);//改变p的大小 
 }

 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()
 {
    int a,b;
    cin>>n;//输入n 
    string s1,s2;//s1:move |pile;  s2:onto |over;
    for(int i=0;i<n;i++) pile[i].push_back(i);//pushback向第i堆木块尾部添加元素 i 
    while(cin>>s1>>a>>s2>>b)//输入指令 ：move a onto b;move a over b;pile aonto b;pile a over b;  
    {
        int pa,pb,ha,hb;
        find_block(a,pa,ha);//找到a所在的堆pa和高度ha 
        find_block(b,pb,hb);
        if(pa==pb) continue;//a,b在同一堆，非法指令 
        if(s2=="onto") clear_above(pb,hb);//onto命令，将b上方木块全部归位 
        if(s1=="move") clear_above(pa,ha);//move命令，将a上方木块全部归位 
        pile_onto(pa,ha,pb);//把a摞在b上方 
     }
     print();
     return 0;
 }