STL(Standard Template Library),即标准模板库。下面是紫书中的几个相关例题,记录学习一下。
1. 排序与检索
algorithm头文件中的sort可以给任意对象排序,包括内置类型和自定义类型,前提是类型定义了“<”运算符。排序之后可以用lower_bound查找大于或等于x的第一个位置。待排序/查找的元素可以放在数组里,也可以放在vector里。还有一个unique函数可以删除有序数组中的重复元素。
例题5-1 大理石在哪儿
现有N个大理石,每个大理石上写了一个非负整数。首先把各数从小到大排序,然后回答Q个问题。每个问题问,是否有一个大理石写着某个整数x,如果是,
还要回答哪个大理石上写着x。排序后的大理石从左到右编号为1~N。(在样例中,为了节约篇幅,所有大理石上的数合并到一行,所有问题也合并到一行)。
样例输入:
4 1
2 3 5 1
5
5 2
1 3 3 3 1
2 3
样例输出:
CASE# 1:
5 found at 4
CASE# 2:
2 not found
3 found at 3
不知道是我理解能力的问题,还是题目本身表达的就不清楚,半天没看懂什么意思QAQ,后来看样例输入才明白,害,,,大致意思就是输入一组数和当前属于的循环次数,然后排序,再查找输入的要找的数的位置。使用algorithm头文件中的sort和lower_bound就可以完成了。lower_bound的作用是查找“大于或等于x的第一个位置”。
#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn=10000;
int main(){
int n,q,x,a[maxn],kase=0;
while(scanf("%d%d",&n,&q)==2 && n){
printf("CASE# %d\n",++kase);
for(int i=0;i<n;i++)
scanf("%d",&a[i]);
sort(a,a+n);//排序
while(q--){
scanf("%d",&x);
int p=lower_bound(a,a+n,x)-a;//在已排序数组a中寻找x
if(a[p]==x)
printf("%d found at %d\n",x,p+1);
else
printf("%d not found\n",x);
}
}
return 0;
}
2. 不定长数组:vector
vector就是一个不定长数组。例如,若a是一个vector,可以用a.size()读取它的大小,a.resize()改变大小,a.push_back()向尾部添加元素,a.pop_back()删除最后一个元素。
vector是一个模板类,所以需要用vector<int> a
或者vector<double> a
这样的方式来声明一个vector。vector<int>
是一个类似于int a[]
的整数数组,而vector<string>
就是一个类似string a[]
的字符串数组。vector可以直接赋值,还可以作为函数的参数或者返回值,而无须像传递数组那样另外用一个变量指定元素个数。
例题5-2 木块问题
从左到右有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在同一堆的指合是非法指合,应当忽略。
分析:
每个木块堆的高度不确定,所以用vector来保存很合适;而木块堆的个数不超过n,所以用一个数组来存就好了。
代码如下:
#include <cstdio>
#include <cstring>
#include <vector>
#include <iostream>
using namespace std;
const int maxn=30;
int n;
vector<int> pile[maxn];//每个pile[i]是一个vector
//找木块a所在的pile和height,以引用的形式返回调用者
void find_block(int a,int& p,int& h){
for(p=0;p<n;p++)
for(h=0;h<pile[p].size();h++)
if(pile[p][h]==a) return ;
}
//把第p堆高度为h的木块上方法所有木块移回原位
void clear_above(int p,int h){
for(int i=h+1;i<pile[p].size();i++){
int b=pile[p][i];
pile[b].push_back(b);//把木块b放回原位
}
pile[p].resize(h+1);//pile只应保留下标0~h的元素
}
//把第p堆高度为h及其上方的木块整体移动到p2堆的顶部
void pile_onto(int p,int h,int p2){
for(int i=h;i<pile[p].size();i++)
pile[p2].push_back(pile[p][i]);
pile[p].resize(h);
}
void print(){
for(int i=0;i<n;i++){
printf("%d:",i);