首页 > 代码库 > 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(一)