本文最后更新于 665 天前,其中的信息可能已经有所发展或是发生改变。
STL 常用操作
| #include <vector> |
| vector<int> a; |
| vector<int> b[233]; |
| struct rec{…}; |
| vector<rec> c; |
a.push_back(x)
把元素x
插入到vector a
的尾部。
a.pop_back()
删除vector a
的最后一个元素。
| vector<int> a; |
| a.push_back(1); |
| a.push_back(2); |
| a.pop_back(); |
size
函数返回vector
的实际长度(包含的元素个数),empty
函数返回一个bool
类型,表明vector
是否为空。二者的时间复杂度都是 O(1)。
| vector<int> a; |
| a.push_back(1); |
| a.push_back(2); |
| |
| cout << a.size() << endl; |
| |
| a.pop_back(); |
| if(a.empty()) cout << "a is empty !" << endl; |
| |
| a.pop_back(); |
| if(a.empty()) cout << "a is empty !" << endl; |
Tips:
利用clear
函数把vector
清空。
| vector<int> a; |
| a.push_back(1); |
| a.push_back(2); |
| |
| cout << a.size() << endl; |
| |
| a.clear(); |
| if(a.empty()) cout << "a is empty !" << endl; |
迭代器就像STL
容器的“指针”,可以用星号*
操作符解除引用。
一个保存int
的vector
的迭代器声明方法为:
| vector<int>::iterator it; |
vector
的迭代器是“随机访问迭代器”,可以把vector
的迭代器与一个整数相加减,其行为和指针的移动类似。可以把vector
的两个迭代器相减,其结果也和指针相减类似,得到两个迭代器对应下标之间的距离。
接下来引入两个迭代器 begin/end
,begin
函数返回指向vector
中第一个元素的迭代器。例如a
是一个非空的vector
,则*a.begin()
与a[0]
的作用相同。
| vector<int> a; |
| for(int i = 1; i <= 10; i ++) a.push_back(i); |
| vector<int>::iterator it = a.begin(); |
| cout << *it << endl; |
所有的容器都可以视作一个“前闭后开”的结构,end
函数返回vector
的尾部,即第 n
个元素再往后的“边界”。*a.end()
与a[n]
都是越界访问,其中n = a.size()
。
| vector<int> a; |
| for(int i = 1; i <= 10; i ++) a.push_back(i); |
| |
| vector<int>::iterator it = a.end() - 1; |
| cout << *it << endl; |
此外 vector
支持利用 []
直接访问元素。
| vector<int> a; |
| for(int i = 1; i <= 10; i ++) a.push_back(i); |
| |
| |
| |
| for(int i = 0; i < a.size(); i ++) cout << a[i] << ' '; |
特别地,可以利用front
函数返回vector
的第一个元素,等价于*a.begin()
和a[0]
。利用back
函数返回vector
的最后一个元素,等价于*--a.end()
和a[a.size() – 1]
。
| vector<int> a; |
| for(int i = 1; i <= 10; i ++) a.push_back(i); |
| |
| cout << a.front() << endl; |
| cout << *a.begin() << endl; |
| cout << a[0] << endl; |
| |
| cout << a.back() << endl; |
| cout << *-- a.end() << endl; |
| cout << a[a.size() - 1] << endl; |
| #include <queue> |
| queue<int> q; |
| struct rec{…}; queue<rec> q; |
| priority_queue<int> q; |
| priority_queue<int, vector<int>, greater<int>> q; |
| priority_queue<pair<int, int>> q; |
循环队列queue
push
将元素插入到队尾,pop
将元素从队头弹出。
| queue<int> a; |
| a.push(1); |
| a.push(2); |
| a.pop(); |
优先队列priority_queue
push
将元素插入堆,pop
删除堆顶元素。
| priority_queue<int> a; |
| a.push(1); |
| a.push(2); |
| a.pop(); |
queue
和 priority_queue
不支持随机访问迭代器。
循环队列queue
| queue<int> a; |
| for(int i = 1; i <= 10; i ++) a.push(i); |
| |
| cout << a.front() << endl; |
| cout << a.back() << endl; |
优先队列priority_queue
| priority_queue<int> a; |
| for(int i = 1; i <= 10; i ++) a.push(i); |
| |
| cout << a.top() << endl; |
| #include <dequeue> |
| [] |
| begin/end |
| front/back |
| push_back |
| push_front |
| pop_back |
| pop_front |
| clear |
双端队列deque
是一个支持在两端高效插入或删除元素的连续线性存储空间。它就像是vector
和queue
的结合。与vector
相比,deque
在头部增删元素仅需要 O(1) 的时间;与queue
相比,deque
像数组一样支持随机访问。
用的不多,对应操作的效率不如 vector
和 queue
。
| #include <set> |
| set<int> s; |
| struct rec{…}; set<rec> s; |
| multiset<double> s; |
头文件set
主要包括set
和multiset
两个容器,分别是“有序集合”和“有序多重集合”,即前者的元素不能重复,而后者可以包含若干个相等的元素。set
和multiset
的内部实现是一棵红黑树,它们支持的函数基本相同。
a.insert(x)
把一个元素x插入到集合a
中,时间复杂度为 O(logn)。
在set
中,若元素已存在,则不会重复插入该元素,对集合的状态无影响。
set
和multiset
的迭代器称为“双向访问迭代器”,不支持“随机访问”,支持星号*
解除引用,仅支持++
和--
两个与算术相关的操作。
一个保存int
的set
的迭代器声明方法为:
接下来亦引入 begin/end
这俩个特殊的迭代器,分别返回集合的首、尾迭代器,时间复杂度均为 O(1)。
a.begin()
是指向集合中最小元素的迭代器。a.end()
是指向集合中最大元素的下一个位置的迭代器。换言之,就像vector
一样,是一个“前闭后开”的形式。因此-- a.end()
是指向集合中最大元素的迭代器。
| set<int> a; |
| for(int i = 10; i >= 1; i --) a.insert(i); |
| |
| cout << *a.begin() << endl; |
| cout << * -- a.end() << endl; |
| |
| |
| for(set<int>::iterator it = a.begin(); it != a.end(); it ++) cout << *it << endl; |
利用a.find(x)
在集合a
中查找等于x
的元素,并返回指向该元素的迭代器。若不存在,则返回a.end()
。时间复杂度为 O(1)。
| set<int> a; |
| |
| for(int i = 10; i >= 1; i --) a.insert(i); |
| |
| set<int>::iterator it = a.find(11); |
| if(it != a.end()) cout << "YES" << endl; |
| else cout << "NO" << endl; |
另外提供a.count(x)
返回集合a
中等于x
的元素个数,时间复杂度为 O(k+logn),其中 k
为元素x
的个数。
| set<int> a; |
| |
| for(int i = 10; i >= 1; i --) a.insert(i); |
| |
| cout << a.count(11) << endl; |
提供erase
方法进行删除操作。
设it
是一个迭代器,a.erase(it)
从a
中删除迭代器it
指向的元素,时间复杂度为 O(logn)。
设x
是一个元素,a.erase(x)
从a
中删除所有等于x
的元素,时间复杂度为 O(k+logn),其中 k
是被删除的元素个数。
| #include <map> |
| map<key_type, value_type> name; |
| |
| |
| map<long long, bool> vis; |
| map<string, int> hash; |
| map<pair<int, int>, vector<int>> test; |
map
容器是一个键值对key-value
的映射,其内部实现是一棵以key
为关键码的红黑树。map
的key
和value
可以是任意类型,其中key
必须定义小于号运算符。
size/empty/clear/begin/end
均于 set
类似,但对于 insert/erase
的参数为 pair<key_type, value_type>
。
| map<int, int> a; |
| a.insert({1, 1}); |
| a.insert({2, 1}); |
特别的,对于 find
,a.find(x)
在变量名为 a
的 map
种,查找 key
为 x
的二元组。
| map<int, int> a; |
| a.insert({1, 1}); |
| a.insert({2, 1}); |
| |
| map<int, int>::iterator it = a.find(2); |
| if(it != a.end()){ |
| cout << "YES" << endl; |
| cout << (*it).first << ' ' << (*it).second << endl; |
| } |
| else cout << "NO" << endl; |
除了和 set
一样利用迭代器进行访问,map
也支持 []
操作符进行访问。我们可以很方便地通过 a[key]
来得到key
对应的value
,还可以对a[key]
进行赋值操作,改变key
对应的value
。
| map<int, int> a; |
| a.insert({1, 1}); |
| a.insert({2, 1}); |
| |
| cout << a[2] << endl; |
| a[2] = 100; |
| cout << a[2] << endl; |
| |
| |
| for(map<int, int>::iterator it = a.begin(); it != a.end(); it ++) cout << (*it).first << ' ' << (*it).second << endl; |
此外,C++11
提供了更方便的遍历 set/map
的方式:
| map<int, int> a; |
| a.insert({1, 1}); |
| a.insert({2, 1}); |
| |
| for(auto &p : a) cout << p.first << ' ' << p.second << endl; |
| |
| set<int> b; |
| b.insert(1); |
| b.insert(2); |
| |
| for(auto &p : b) cout << p << endl; |
| #include <string> |
| string(const char* s); |
| string(const string& str); |
| string(int n, char c); |
string
可以视为一个动态字符数组使用。
| string str1 = "abcdefghigklmn" ; |
| |
| string str2 = str1.substr(2,5) ; |
| |
| str1.reverse(s.begin(), s.end()); |
| |
| int p1 = str1.find(str); |
| |
| int p2 = str1.find(str,3); |
| |
| int p3 = str1.rfind(str); |
| |
| int p4 = str1.rfind(str,3); |
| |
| string str3 = "1234.567" ; |
| |
| double nums = atof(str3.c_str()); |
| |
| int nums = atoi(str3.c_str()); |
| string s = "abcd"; |
| cout << s[0] << endl; |
| cout << s.front() << endl; |
| cout << *s.begin() << endl; |
| |
| cout << s[s.size() - 1] << endl; |
| cout << s.back() << endl; |
| cout << * -- s.end() << endl; |
pair
是标准库中定义的一个类模板。用于将两个变量关联在一起,组成一个“对”,而且两个变量的数据类型可以是不同的。
| pair<int, int> a; |
| pair<int, int> b = {1, 2}; |
| pair<int, double> c; c.first = 1, c.second = 2.333; |
| pair<int, int> d[100]; |
在 pair
里已经预先定义了所有的比较运算符,包括 <
、>
、<=
、>=
、==
、!=
。当然,这需要组成 pair
的两个变量所属的数据类型定义了 ==
和/或 <
运算符。
其中,<
、>
、<=
、>=
四个运算符会先比较两个 pair
中的第一个变量,在第一个变量相等的情况下再比较第二个变量。
| pair<int, string> name[10]; |
| |
| name[0] = {3, "lys"}; |
| name[1] = {2, "hfcj"}; |
| name[2] = {3, "wjq"}; |
| |
| sort(name, name + 3); |
| |
| for(int i = 0; i < 3; i ++){ |
| cout << name[i].first << ' ' << name[i].second << endl; |
| } |
Tips:
很多情况下,map
中存储的键值对通过 pair
向外暴露。
| int a[100] = {0}; |
| for(int i = 0; i < 10; i ++) a[i] = i + 1; |
| |
| sort(a, a + 10); |
| |
| int pos1 = lower_bound(a, a + 10, 7) - a; |
| int pos2 = upper_bound(a, a + 10, 7) - a; |
| |
| cout << pos1 << ' ' << a[pos1] << endl; |
| cout << pos2 << ' ' << a[pos2] << endl; |
| |
| sort(a, a + 10, greater<int>()); |
| |
| int pos3 = lower_bound(a, a + 10, 7, greater<int>()) - a; |
| int pos4 = upper_bound(a, a + 10, 7, greater<int>()) - a; |
| |
| cout << pos3 << ' ' << a[pos3] <<endl; |
| cout << pos4 << ' ' << a[pos4] <<endl; |
Tips:
- 适用
vector
返回对应元素的迭代器。
- 适用
map
返回对应 key
的迭代器。
- 在一般的数组和
vector
里,这两个函数的时间复杂度均为 O(logn),但在 set/map
等关联式容器中,直接调用 lower_bound(s.begin(),s.end(),val)
的时间复杂度是 O(n) 的。
- 为此
set/map
等关联式容器中已经封装了 lower_bound
等函数(像 s.lower_bound(val)
这样),这样调用的时间复杂度是 O(logn) 的。
作用:
- 将当前排列更改为全排列中的下一个排列。
- 如果当前排列已经是全排列中的最后一个排列(元素完全从大到小排列),函数返回
false
并将排列更改为全排列中的第一个排列(元素完全从小到大排列);否则,函数返回 true
。
| int a[] = {1, 2, 3, 4, 5}; |
| do{ |
| for(int i = 0; i < 5; i ++) cout << a[i] << " "; |
| cout << endl; |
| }while(next_permutation(a, a + 5)); |
| vector<int> a = {1, 2, 3, 4, 5}; |
| do{ |
| for(int i = 0; i < a.size(); i ++) cout << a[i] << " "; |
| cout << endl; |
| }while(next_permutation(a.begin(), a.end())); |