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();
}