初学acmer--读《算法竞赛入门经典》笔记 p110-111 Uva101

题目：给你n个方块，有四种操作：

            1.move a onto b，把a和b上面的方块都放回原来位置，然后把a放到b上面；

            2.move a over b，把a上面的放回原处，然后把a放在b所在的方块堆的上面；

            3.pile a onto b，把b上面的放回原来位置，然后把a和a上面的方块整体放到b上面；

            4.pile a over b，把a和a上面的方块整体放到b所在堆的上面。

样例输入：

10 
move 9 onto 1 
move 8 over 1 
move 7 over 1 
move 6 over 1 
pile 8 over 6 
pile 8 over 5 
move 2 over 1 
move 4 over 9 
quit 

样例输出：

0: 0 
1: 1 9 2 4 
2: 
3: 3 
4: 
5: 5 8 7 6 
6: 
7: 
8: 
9:

分析：这一题在理解了题意后，选择正确合适的数据结构就显得尤为重要。由于每个木块堆的高度不确定，所以用vector来保存就很合适，而木块的堆数不止一个且不确定，所以可以定义vector的数组，即定义vector<int> pile[maxn]。此时vector就像一个二维数组，只是第一维的大小是固定的，但是第二维的大小是可以改变的。vector是一个不定长的数组，可以用clear()来清空，resize（）来改变长度，用push_back()和pop_back()在尾部添加和删除元素，用empty（）测试是否为空，vector之间可以直接赋值或者作为函数的返回值，和int,char等普通数据类型一样。

下面贴代码：

#include<iostream>
#include<cstdio>
#include<string>
#include<vector>
using namespace std;
const int maxn=30;
int n;
 vector<int> p[maxn];              //①
void find_blocks(int a,int &pa,int &ha)    // ③
{
    for(pa=0;pa<n;pa++)
    {
        for(ha=0;ha<p[pa].size();ha++)
        {
            if(a==p[pa][ha]) return;
        }
    }
}
void clear_blocks(int p1,int h)
{
    for(int i=h+1;i<p[p1].size();i++)
    {
        int t=p[p1][i];
        p[t].push_back(t);
    }
    p[p1].resize(h+1);
}
void pile_onto(int pa,int pb,int ha)
{
    for(int i=ha;i<p[pa].size();i++)
    {
        p[pb].push_back(p[pa][i]);
    }
    p[pa].resize(ha);
}
int  main()
{
    int a,b;
    cin>>n;
    string s1,s2;
    for(int i=0;i<n;i++)
    {
        p[i].push_back(i);
    }
    while(cin>>s1>>a>>s2>>b)
    {
        int pa,pb,ha,hb;
        find_blocks(a,pa,ha);
        find_blocks(b,pb,hb);
        if(pa==pb) continue;
        if(s2=="onto") clear_blocks(pb,hb);
        if(s1=="move") clear_blocks(pa,ha);       //②
        pile_onto(pa,pb,ha);
    }
    for(int i=0;i<n;i++)
    {
        cout<<i<<':';
        for(int j=0;j<p[i].size();j++)
        {
            cout<<p[i][j];
        }
        cout<<endl;
    }
}

ps:①：由于vector在main函数和其他函数都要用到，所以将其定义为全局比较方便。

②：题目中一共4种指令，如果对它们都写一个代码，就很麻烦，更好的方法就是提取指令之间的共同点，将相同的操作用一个函数来表示，从而减少代码量。

③：由于要改变的值不止一个，所以可以采用引用的方式返回调用者，不过要注意的是，在函数中for语句中不能再习惯性的打上(int pa....)