ACM基础之不定长数组vector与应用举例

vector是一个不定长数组，使用时需要先加载头文件：#include<vector>
这里列举几个最常用的操作：

vector<int>a;								//初始化一个vector
int n = a.size();							//读取vector的大小
a.resize(int n);							//改变a的大小
a.push_back();								//向vector的尾部添加元素
a.pop_back();								//删除vector最后一个元素

应用举例：
The Blocks Problem(Uva 101)
【题目描述】
从左到右有n个木块，编号为0~n-1，要去模拟一下4种操作（下面的a和b都是木块编号）

• move a onto b：把a和b上方的木块全部归位，然后把a摞在b上面；
• move a over b：把a上方的木块全部归位，然后把a放在b所在木块堆的顶部；
• pile a onto b：把b上方的木块全部归为，然后把a及上面的木块整体摞在b上方；
• pile a over b：把a以及上面的木块整体摞在b所在木块堆的顶部

遇到 quit时终止一组数据。a和b在同一堆的指令是非法指令，应当忽略。
所有操作结束后，输出每个位置的木块列表，按照从底部到顶部的顺序排列
【Sample Input】
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
【Sample Output】
0: 0
1: 1 9 2 4
2:
3: 3
4:
5: 5 8 7 6
6:
7:
8:
9:
【分析】
每个木块堆的高度不确定，所以用vector保存很合适。，而木块堆的个数不超过n，所以用一个数组来存就可以了
这里要提及的是：这道题有两种做法，因为这道题中，有四种操作，分别为：move a onto b、move a over b、pile a onto b、pile a over b，所以第一种想法是对四种操作每种写一个独立的函数，这当然是可以的，但是如果完全独立地处理各个指令，指令之间有很多操作是重复的，这样代码就会变得冗长易错。还有一种想法是提取这四种指令的共同点，编写函数减少重复的代码。
对这四种操作加以分析我们不难发现，输入包括四个部分：s1(string), a(int), s2(string), b(int),当s1 == "move"时，a上面所有的木块都要归位，s1 == "pile"时，a上面所有的木块不归位，s2 == "onto"时，b上面所有的木块都要归位，s2 == "over"时，b上面所有的木块不归位，这四种操作可以总结为：在进行完以上归位或不归位操作后，将a及以上的所有木块都放在b的木块堆的上方。
这个思想很重要，读者要仔细理解。

在以上思想的支撑下，代码如下：

#include<iostream>
#include<vector>
using namespace std;
const int maxn = 30;
vector<int> pile[maxn];
int n;                  //n个木块堆

void find_block(int a, int &p, int &h)
{//寻找编号为n的木块在第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][h];
        pile[b].push_back(b);           //把木块b放回原位
    }
    pile[p].resize(h+1);                //pile只应保留下表0~h的元素
}

void pile_onto(int p, int h, int p2)
{//将第p堆木块h及以上的所有木块都移动到p2的顶部
    for(int i=h; i<pile[p].size(); i++)
    {
        int b = pile[p][i];
        pile[p2].push_back(b);
    }
    pile[p].resize(h);
}

void print()
{
    for(int i=0; i<n; i++)
    {
        cout<<i<<":";
        for(int j=0; j<pile[i].size(); j++)
        {
            cout<<" "<<pile[i][j];
        }
        cout<<endl;
    }
}

int main()
{
    cin>>n;
    string s1, s2;
    int a, b;
    for(int i=0; i<n; i++) pile[i].push_back(i);
    while(cin>>s1)
    {
        if(s1 == "quit") break;
        cin>>a>>s2>>b;
        int pa, ha, pb, hb;
        find_block(a, pa, ha);
        find_block(b, pb, hb);
        if(pa == pb) continue;              //非法指令，跳过
        if(s1 == "move")
        {
            clear_above(pa, ha);
        }
        if(s2 == "onto")
        {
            clear_above(pb, hb);
        }
        pile_onto(pa, ha, pb);
    }
    print();
}