首页 > 代码库 > ACM竞赛常用STL(一)
ACM竞赛常用STL(一)
全排列函数next_permutation
STL 中专门用于排列的函数(可以处理存在重复数据集的排列问题)
头文件:#include <algorithm>
using namespace std;
调用: next_permutation(start, end);
注意:函数要求输入的是一个升序排列的序列的头指针和尾指针.
用法:
1 // 数组 2 int a[N]; 3 sort(a, a+N); 4 next_permutation(a, a+N); 5 // 向量 6 vector<int> ivec; 7 sort(ivec.begin(), ivec.end()); 8 next_permutation(ivec.begin(), ivec.end()); 9 例子:10 vector<int> myVec;11 // 初始化代码12 sort(myVec.begin(),myVec.end());13 do{14 for (i = 0 ;i < size;i ++ ) cout << myVec[i] << " \t " ;15 cout << endl;16 }while (next_permutation(myVec.begin(), myVec.end()));
ACM/ICPC 竞赛之STL--pair
STL 的<utility>头文件中描述了一个看上去非常简单的模板类pair,用来表示一个二元组或元素对,并提供了按照字典序对元素对进行大小比较的比较运算符模板函数。
例如,想要定义一个对象表示一个平面坐标点,则可以:
pair<double, double> p1;cin >> p1.first >> p1.second;pair 模板类需要两个参数:首元素的数据类型和尾元素的数据类型。pair 模板类对象有两个成员:first 和second,分别表示首元素和尾元素。
在<utility>中已经定义了pair 上的六个比较运算符:<、>、<=、>=、==、!=,其规则是先比较first,first 相等时再比较second,这符合大多数应用的逻辑。当然,也可以通过重载这几个运算符来重新指定自己的比较逻辑。除了直接定义一个pair 对象外,如果需要即时生成一个pair 对象,也可以调用在<utility>中定义的一个模板函数:make_pair。make_pair 需要两个参数,分别为元素对的首元素和尾元素。
在题1067--Ugly Numbers 中,就可以用pair 来表示推演树上的结点,用first 表示结点的值,用second 表示结点是由父结点乘以哪一个因子得到的。
1 #include <iostream> 2 #include <queue> 3 using namespace std; 4 typedef pair<unsigned long, int> node_type; 5 int main() 6 { 7 unsigned long result[1500]; 8 priority_queue< node_type, vector<node_type>, 9 greater<node_type> > Q;10 Q.push( make_pair(1, 2) );11 for (int i=0; i<1500; i++)12 {13 node_type node = Q.top(); Q.pop();14 switch(node.second)15 {16 case 2: Q.push( make_pair(node.first*2, 2) );17 case 3: Q.push( make_pair(node.first*3, 3) );18 case 5: Q.push( make_pair(node.first*5, 5) );19 }20 result[i] = node.first;21 }22 int n;23 cin >> n;24 while (n>0)25 {26 cout << result[n-1] << endl;27 cin >> n;28 }29 return 0;30 }
ACM/ICPC 竞赛之STL--vector
在STL 的<vector>头文件中定义了vector(向量容器模板类),vector容器以连续数组的方式存储元素序列,可以将vector 看作是以顺序结构实现的线性表。当我们在程序中需要使用动态数组时,vector 将会是理想的选择,vector 可以在使用过程中动态地增长存储空间。
vector 模板类需要两个模板参数,第一个参数是存储元素的数据类型,第二个参数是存储分配器的类型,其中第二个参数是可选的,如果不给出第二个参数,将使用默认的分配器。
下面给出几个常用的定义vector 向量对象的方法示例:38
vector<int> s;
定义一个空的vector 对象,存储的是int 类型的元素。
vector<int> s(n);定义一个含有n 个int 元素的vector 对象。
vector<int> s(first, last);定义一个vector 对象,并从由迭代器first 和last 定义的序列[first,last)中复制初值。
vector 的基本操作有:
s[i]直接以下标方式访问容器中的元素。
s.front() 返回首元素。
s.back() 返回尾元素。
s.push_back(x)向表尾插入元素x。
s.size() 返回表长。
s.empty() 当表空时,返回真,否则返回假。
s.pop_back() 删除表尾元素。
s.begin() 返回指向首元素的随机存取迭代器。
s.end() 返回指向尾元素的下一个位置的随机存取迭代器。
s.insert(it, x) 向迭代器it 指向的元素前插入新元素val。
s.insert(it, n, x)向迭代器it 指向的元素前插入n 个x。
s.insert(it, first, last)将由迭代器first 和last 所指定的序列[first, last)插入到迭代器it
指向的元素前面。
s.erase(it)删除由迭代器it 所指向的元素。
s.erase(first, last)删除由迭代器first 和last 所指定的序列[first, last)。
s.reserve(n)预分配缓冲空间,使存储空间至少可容纳n 个元素。
s.resize(n)改变序列的长度,超出的元素将会被删除,如果序列需要扩展(原空间小于n),元素默认值将填满扩展出的空间。
s.resize(n, val)改变序列的长度,超出的元素将会被删除,如果序列需要扩展(原空间小于n),将用val 填满扩展出的空间。
s.clear()删除容器中的所有的元素。
s.swap(v)将s 与另一个vector 对象v 进行交换。
s.assign(first, last)将序列替换成由迭代器first 和last 所指定的序列[first, last)。[first, last)不能是原序列中的一部分。要注意的是,resize 操作和clear 操作都是对表的有效元素进行的操作,但并不一定会改变缓冲空间的大小。另外,vector 还有其他一些操作如反转、取反等,不再一下列举。vector 上还定义了序列之间的比较操作运算符(>, <, >=, <=, ==, !=),
可以按照字典序比较两个序列。还是来看一些示例代码。输入个数不定的一组整数,再将这组整数按倒序输出,
如下所示:
1 #include <iostream> 2 #include <vector> 3 using namespace std; 4 int main() 5 { 6 vector<int> L; 7 int x; 8 while (cin>>x) L.push_back(x); 9 for (int i=L.size()-1; i>=0; i--) 10 cout << L[i] << " ";11 cout << endl;12 return 0;13 }
ACM/ICPC 竞赛之STL--iterator 简介
iterator(迭代器)是用于访问容器中元素的指示器,从这个意义上说,iterator(迭代器)相当于数据结构中所说的“遍历指针”,也可以把iterator(迭代器)看作是一种泛化的指针。STL 中关于iterator(迭代器)的实现是相当复杂的,这里我们暂时不去详细讨论关于iterator(迭代器)的实现和使用,而只对iterator(迭代器)做一点简单的介绍。
简单地说,STL 中有以下几类iterator(迭代器):
输入iterator(迭代器),在容器的连续区间内向前移动,可以读取容器内任意值;输出iterator(迭代器),把值写进它所指向的容器中;前向iterator(迭代器),读取队列中的值,并可以向前移动到下一位置(++p,p++);双向iterator(迭代器),读取队列中的值,并可以向前向后遍历容器;随机访问iterator(迭代器), 可以直接以下标方式对容器进行访问,vector 的iterator(迭代器)就是这种iterator(迭代器);流iterator(迭代器),可以直接输出、输入流中的值;每种STL 容器都有自己的iterator(迭代器)子类,下面先来看一段简单的示例代码:
1 #include <iostream> 2 #include <vector> 3 using namespace std; 4 main() 5 { 6 vector<int> s; 7 for (int i=0; i<10; i++) 8 s.push_back(i); 9 for (vector<int>::iterator it=s.begin(); it!=s.end();it++)10 cout << *it << " ";11 cout << endl;12 return 1;13 }
vector 的begin()和end()方法都会返回一个vector::iterator 对象,分别指向vector 的首元素位置和尾元素的下一个位置(我们可以称之为结束标志位置)。对一个iterator(迭代器)对象的使用与一个指针变量的使用极为相似,或者可以这样说,指针就是一个非常标准的iterator(迭代器)。再来看一段稍微特别一点的代码:
1 #include <iostream> 2 #include <vector> 3 using namespace std; 4 main() 5 { 6 vector<int> s; 7 s.push_back(1); 8 s.push_back(2); 9 s.push_back(3);10 copy(s.begin(), s.end(), ostream_iterator<int>(cout, ""));11 cout <<endl;12 return 1;13 }
这段代码中的copy 就是STL 中定义的一个模板函数,copy(s.begin(),s.end(), ostream_iterator<int>(cout, " "));的意思是将由s.begin()至s.end()(不含s.end())所指定的序列复制到标准输出流cout 中,用" "作为每个元素的间隔。也就是说,这句话的作用其实就是将表中的所有内容依次输出。iterator(迭代器)是STL 容器和算法之间的“胶合剂”,几乎所有的STL 算法都是通过容器的iterator(迭代器)来访问容器内容的。只有通过有效地运用iterator(迭代器),才能够有效地运用STL 强大的算法功能。
ACM/ICPC 竞赛之STL--string
字符串是程序中经常要表达和处理的数据,我们通常是采用字符数组或字符指针来表示字符串。STL 为我们提供了另一种使用起来更为便捷的字符串的表达方式:string。string 类的定义在头文件<string>中。string 类其实可以看作是一个字符的vector,vector 上的各种操作都可以适用于string,另外,string 类对象还支持字符串的拼合、转换等操作。下面先来看一个简单的例子:
1 #include <iostream> 2 #include <string> 3 using namespace std; 4 int main() 5 { 6 string s = "Hello! ", name; 7 cin >> name; 8 s += name; 9 s += ‘!‘;10 cout << s << endl;11 return 0;12 }
再以题1064--Parencoding 为例,看一段用string 作为容器,实现由P
代码还原括号字符串的示例代码片段:
1 int m; 2 3 cin >> m; // P 编码的长度 4 5 string str; // 用来存放还原出来的括号字符串 6 7 int leftpa = 0; // 记录已出现的左括号的总数 8 9 for (int j=0; j<m; j++)10 {11 12 int p;13 14 cin >> p;15 16 for (int k=0; k<p-leftpa; k++) str += ‘(‘;17 18 str += ‘)‘;19 20 leftpa = p;21 22 }
ACM/ICPC 竞赛之STL--stack/queue
stack(栈)和queue(队列)也是在程序设计中经常会用到的数据容器,STL为我们提供了方便的stack(栈)的queue(队列)的实现。39准确地说,STL 中的stack 和queue 不同于vector、list 等容器,而是对这些容器的重新包装。这里我们不去深入讨论STL 的stack 和queue 的实现细节,而是来了解一些他们的基本使用。
1、stack
stack 模板类的定义在<stack>头文件中。
stack 模板类需要两个模板参数,一个是元素类型,一个容器类型,但只有元
素类型是必要的,在不指定容器类型时,默认的容器类型为deque。
定义stack 对象的示例代码如下:
stack<int> s1;
stack<string> s2;
stack 的基本操作有:
入栈,如例:s.push(x);
出栈,如例:s.pop();注意,出栈操作只是删除栈顶元素,并不返回该元素。
访问栈顶,如例:s.top()
判断栈空,如例:s.empty(),当栈空时,返回true。
访问栈中的元素个数,如例:s.size()
下面是用string 和stack 写的解题1064--Parencoding 的程序。
1 #include <iostream> 2 #include <string> 3 #include <stack> 4 using namespace std; 5 int main() 6 { 7 int n; 8 cin >> n; 9 for (int i=0; i<n; i++)10 {11 int m;12 cin >> m;13 string str;14 int leftpa = 0;15 for (int j=0; j<m; j++) // 读入P 编码,构造括号字符串16 { 17 int p;18 cin >> p;19 for (int k=0; k<p-leftpa; k++) 20 str += ‘(‘;21 str += ‘)‘;22 leftpa = p;23 }24 stack<int> s;25 for (string::iterator it=str.begin();it!=str.end(); it++) 26 { // 构造M 编码27 if (*it==‘(‘) s.push(1);28 else29 {30 int p = s.top(); s.pop();31 cout << p << " ";32 if (!s.empty()) s.top() += p;33 }34 }35 cout << endl;36 }37 return 0;38 }
2、queue
queue 模板类的定义在<queue>头文件中。stack 模板类很相似,queue 模板类也需要两个模板参数,一个是元素类型,一个容器类型,元素类型是必要的,容器类型是可选的,默认为deque 类型。
定义queue 对象的示例代码如下:
queue<int> q1;
queue<double> q2;
queue 的基本操作有:
入队,如例:q.push(x); 将x 接到队列的末端。
出队,如例:q.pop(); 弹出队列的第一个元素,注意,并不会返回被弹出元素的值。
访问队首元素,如例:q.front(),即最早被压入队列的元素。
访问队尾元素,如例:q.back(),即最后被压入队列的元素。
判断队列空,如例:q.empty(),当队列空时,返回true。
访问队列中的元素个数,如例:q.size()
3、priority_queue
在<queue>头文件中,还定义了另一个非常有用的模板类priority_queue(优先队列)。优先队列与队列的差别在于优先队列不是按照入队的顺序出队,而是按照队列中元素的优先权顺序出队(默认为大者优先,也可以通过指定算子来指定自己的优先顺序)。priority_queue 模板类有三个模板参数,第一个是元素类型,第二个容器类型,第三个是比较算子。其中后两个都可以省略,默认容器为vector,默认算子为less,即小的往前排,大的往后排(出队时序列尾的元素出队)。
定义priority_queue 对象的示例代码如下:
priority_queue<int> q1;
priority_queue< pair<int, int> > q2; // 注意在两个尖括号之间一定要留空格。
priority_queue<int, vector<int>, greater<int> > q3; // 定义小的先出队
priority_queue 的基本操作与queue 相同。
初学者在使用priority_queue 时,最困难的可能就是如何定义比较算子了。如果是基本数据类型,或已定义了比较运算符的类,可以直接用STL 的less算子和greater 算子——默认为使用less 算子,即小的往前排,大的先出队。如果要定义自己的比较算子,方法有多种,这里介绍其中的一种:重载比较运算符。优先队列试图将两个元素x 和y 代入比较运算符(对less 算子,调用x<y,对greater 算子,调用x>y),若结果为真,则x 排在y 前面,y 将先于x 出队,反之,则将y 排在x 前面,x 将先出队。
看下面这个简单的示例:
1 #include <iostream> 2 #include <queue> 3 using namespace std; 4 class T 5 { 6 public: 7 int x, y, z; 8 T(int a, int b, int c):x(a), y(b), z(c){} 9 };10 bool operator < (const T &t1, const T &t2)11 {12 return t1.z < t2.z; // 按照z 的顺序来决定t1 和t2 的顺序13 }14 int main()15 {16 priority_queue<T> q;17 q.push(T(4,4,3));18 q.push(T(2,2,5));19 q.push(T(1,5,4));20 q.push(T(3,3,6));21 while (!q.empty())22 {23 T t = q.top(); q.pop();24 cout << t.x << " " << t.y << " " << t.z << endl;25 }26 return 0; 27 }28 /*输出结果为(注意是按照z 的顺序从大到小出队的):29 3 3 630 2 2 531 1 5 432 4 4 3*/33 34 35 //再看一个按照z 的顺序从小到大出队的例子:36 #include <iostream>37 #include <queue>38 using namespace std;39 class T40 {41 public:42 int x, y, z;43 T(int a, int b, int c):x(a), y(b), z(c)44 {45 }46 };47 bool operator > (const T &t1, const T &t2)48 {49 return t1.z > t2.z;50 }51 int main()52 {53 priority_queue<T, vector<T>, greater<T> > q;54 q.push(T(4,4,3));55 q.push(T(2,2,5));56 q.push(T(1,5,4));57 q.push(T(3,3,6));58 while (!q.empty())59 {60 T t = q.top(); q.pop();61 cout << t.x << " " << t.y << " " << t.z << endl;62 }63 return 0;64 }
输出结果为:
4 4 3
1 5 4
2 2 5
3 3 6
如果我们把第一个例子中的比较运算符重载为:
bool operator < (const T &t1, const T &t2){
return t1.z > t2.z; // 按照z 的顺序来决定t1 和t2 的顺序
}
则第一个例子的程序会得到和第二个例子的程序相同的输出结果。
再回顾一下用优先队列实现的题1067--Ugly Numbers 的代码:
1 #include <iostream> 2 #include <queue> 3 using namespace std; 4 typedef pair<unsigned long int, int> node_type; 5 int main( int argc, char *argv[] ) 6 { 7 unsigned long int result[1500]; 8 priority_queue< node_type, vector<node_type>, 9 greater<node_type> > Q;10 Q.push( make_pair(1, 3) );11 for (int i=0; i<1500; i++)12 {13 node_type node = Q.top();14 Q.pop();15 switch(node.second)16 {17 case 3: Q.push( make_pair(node.first*2, 3) );18 case 2: Q.push( make_pair(node.first*3, 2) );19 case 1: Q.push( make_pair(node.first*5, 1) );20 }21 result[i] = node.first;22 }23 int n;24 cin >> n;25 while (n>0)26 {27 cout << result[n-1] << endl;28 cin >> n;29 }30 return 1;31 }
ACM/ICPC 竞赛之STL--map
在STL 的头文件<map>中定义了模板类map 和multimap,用有序二叉树来存贮类型为pair<const Key, T>的元素对序列。序列中的元素以const Key部分作为标识,map 中所有元素的Key 值都必须是唯一的,multimap 则允许有重复的Key 值。可以将map 看作是由Key 标识元素的元素集合,这类容器也被称为“关联容器”,可以通过一个Key 值来快速确定一个元素,因此非常适合于需要按照Key值查找元素的容器。map 模板类需要四个模板参数,第一个是键值类型,第二个是元素类型,第三个是比较算子,第四个是分配器类型。其中键值类型和元素类型是必要的。map 的基本操作有:
1、定义map 对象,例如:
map<string, int> m;
2、向map 中插入元素对,有多种方法,例如:
m[key] = value;
[key]操作是map 很有特色的操作,如果在map 中存在键值为key 的元素对,
则返回该元素对的值域部分,否则将会创建一个键值为key 的元素对,值域为默认值。所以可以用该操作向map 中插入元素对或修改已经存在的元素对的值域部分。
m.insert( make_pair(key, value) );
也可以直接调用insert 方法插入元素对,insert 操作会返回一个pair,当map 中没有与key 相匹配的键值时,其first 是指向插入元素对的迭代器,其second 为true;若map 中已经存在与key 相等的键值时,其first 是指向该元素对的迭代器,second 为false。
3、查找元素对,例如:
int i = m[key];
要注意的是,当与该键值相匹配的元素对不存在时,会创建键值为key 的元素对。map<string, int>::iterator it = m.find(key);如果map 中存在与key 相匹配的键值时,find 操作将返回指向该元素对的迭代器,否则,返回的迭代器等于map 的end()(参见vector 中提到的begin
和end 操作)。
4、删除元素对,例如:
m.erase(key);删除与指定key 键值相匹配的元素对,并返回被删除的元素的个数。
m.erase(it);删除由迭代器it 所指定的元素对,并返回指向下一个元素对的迭代器。
看一段简单的示例代码:
1 #include<map> 2 #include<iostream> 3 using namespace std; 4 typedef map<int, string, less<int> > M_TYPE; 5 typedef M_TYPE::iterator M_IT; 6 typedef M_TYPE::const_iterator M_CIT; 7 int main() 8 { 9 M_TYPE MyTestMap;10 MyTestMap[3] = "No.3";11 MyTestMap[5] = "No.5";12 MyTestMap[1] = "No.1";13 MyTestMap[2] = "No.2";14 MyTestMap[4] = "No.4";15 M_IT it_stop = MyTestMap.find(2);16 cout << "MyTestMap[2] = " << it_stop->second << endl;17 it_stop->second = "No.2 After modification";18 cout << "MyTestMap[2] = " << it_stop->second << endl;19 cout << "Map contents : " << endl;20 for(M_CIT it = MyTestMap.begin(); it != MyTestMap.end();it++)21 {22 cout << it->second << endl;23 }24 return 0;25 }26 /*程序执行的输出结果为:27 MyTestMap[2] = No.228 MyTestMap[2] = No.2 After modification29 Map contents :30 No.131 No.2 After modification32 No.333 No.434 No.5*/
再看一段简单的示例代码:
1 #include <iostream> 2 #include <map> 3 using namespace std; 4 int main() 5 { 6 map<string, int> m; 7 m["one"] = 1; 8 m["two"] = 2; 9 // 几种不同的insert 调用方法10 m.insert(make_pair("three", 3));11 m.insert(map<string, int>::value_type("four", 4));12 m.insert(pair<string, int>("five", 5));13 string key;14 while (cin>>key)15 {16 map<string, int>::iterator it = m.find(key);17 if (it==m.end())18 {19 cout << "No such key!" << endl;20 }21 else22 {23 cout << key << " is " << it->second << endl;24 cout << "Erased " << m.erase(key) << endl;25 }26 }27 return 0;28 }
ACM竞赛常用STL(一)