C++ STL 容器set和算法for_each小案例

关键词：C++， 容器set，算法for_each， sort

笔者在之前的文章中使用的STL容器都是vector。

现在简单介绍一下另一个容器set。

set容器的特点：

• 内部元素有序排列，新元素插入的位置取决于它的值，查找速度快；
• 插入已有元素的时候，忽略插入，就是说不允许有重复的元素。

set的底层实现笔者不讲了，举个例子说说这个容器的应用。

set经常被用于去重复，一行代码就搞定，很方便。

笔者之前文章中介绍了去除文件中相邻的重复行的方法；

如果文件中某些行的内容相同，但是被别的行隔开了，就不能去掉所有重复的内容;

这就需要先sort。

加上sort的代码如下：

 string i_line ; // each line from file
 vector<string> all_lines ; // container to store all lines
 while ( getline(input_f, i_line) ) { // get each line

 all_lines.push_back(i_line);
 }
 //sort
 sort(all_lines.begin(),all_lines.end());
 //uniq
 vector<string>::iterator ii ;
 for( ii = all_lines.begin() ; ii < (all_lines.end() - 1) ; ++ii ) {
 while ( *ii == * (ii + 1) ) {
 all_lines.erase(ii+1) ;
 }
 // cout << *ii << endl;
 }

这一小段代码，sort之后的7行代码的内容可以用这样一行代码来替代：

set<string> all_sort_uniq_lines(all_lines.begin(), all_lines.end()) ;

是不是很酷呢？

再介绍一个很酷的东西，就是STL的for_each算法。

for_each算法的作用是对某个区间的每个元素都做某种操作。

foreach的定义：

template<class InIt, class Fun>
Fun for_each(InIt first, InIt last, Fun f);

对 [first,last) 中的每个元素 e , 执行 f(e) , 要求 f(e) 不能改变 e 。

我们把一个vector容器的内容都放进set容器之后，

初始化set的时候就会自动做排序和去掉重复的内容。

之后我们需要输出排序和去重复后的每个元素的内容，代码是这样的：

 set<string>::iterator i ;
 for( i = all_sort_uniq_lines.begin() ; i != all_sort_uniq_lines.end() ; ++i ) {
 cout << *i << endl;
 }

这个代码中规中矩，

如果用for_each，就只是这样一行代码：

for_each(all_sort_uniq_lines.begin(),all_sort_uniq_lines.end(), outputString) ;

其中outputString是一个函数，定义如下：

void outputString(string i_s) {
 cout << i_s << endl ;
}

酷不酷？

程序源代码：

#include <iostream>
#include <fstream>
#include <string>
#include <vector>
#include <set>
#include <algorithm>
#include <iterator>
using namespace std;

void outputString(string i_s) {
 cout << i_s << endl ;
}

int main (int argc, char* argv[])
{
 // usage: myuniq -help
 // usage: myuniq file_name
 if (argc > 2 ) {
 cout << "Wrong input option number:" << argc -1 << endl;
 cout << "Please input:myuniq file_name" << endl;
 return 0 ;
 }

 // check -help
 string option_1 = argv[1] ;
 if (option_1 == "-help" ) {
 cout << " usage: myuniq file_name " << endl ;
 return 0 ;
 }

 // open file
 // use argv[1] directly, cannot use string,
 // because ifstream input shoud be char *
 ifstream input_f(argv[1]) ; // input argv 1
 if ( !input_f ) { // check if input file can be opened
 cout << "Error: the input file can not be opened" <<endl ;
 return 0 ;
 }

 string i_line ; // each line from file
 vector<string> all_lines ; // container to store all lines
 while ( getline(input_f, i_line) ) { // get each line

 all_lines.push_back(i_line);
 }

 //sort + uniq using set
 //
 set<string> all_sort_uniq_lines(all_lines.begin(), all_lines.end()) ;

 for_each(all_sort_uniq_lines.begin(),all_sort_uniq_lines.end(), outputString) ;

 input_f.close() ;

 return 0 ;
}

测试文件test.txt:

12 24 45 9 48 5


12 24 45 9 48 5
12 24 45 9 48 5
12 24 45 9 48 5
12 24 65 c 48 5
12 24 45 b 48 5
12 24 45 a 48 5
12 24 45 9 48 5



12 24 75 f 48 5
12 24 75 f 48 5
12 24 75 f 48 5
12 24 75 f 48 5
12 24 45 d 48 5
12 24 95 9 48 5
12 24 45 9 48 5


12 24 35 9 48 5
12 24 25 9 48 5
12 24 25 9 48 5

测试结果：

g++ file_sort_uniq.set.for_each.cpp -o file_sort_uniq.set.for_each
./file_sort_uniq.set.for_each test.txt

12 24 25 9 48 5
12 24 35 9 48 5
12 24 45 9 48 5
12 24 45 a 48 5
12 24 45 b 48 5
12 24 45 d 48 5
12 24 65 c 48 5
12 24 75 f 48 5
12 24 95 9 48 5