首页 > 代码库 > C++运算符重载的妙用
C++运算符重载的妙用
运算符重载(Operator overloading)是C++重要特性之一,本文将介绍运算符重载在C++标准库里的妙用。如果你对C++的运算符重载掌握的游刃有余,那就无需继续往下看了。
运算符重载带来的好处就是——让代码变得简洁。下面将展示几个标准库因使用运算符重载而是代码简洁的实例。
Hello, World与operator<<
刚学C++时看到的第一个C++程序就是Hello World,它当时长得这样:
#include <iostream> using namespace std; int main(int argc, char *argv[]) { cout << "Hello, world!" << endl; return 0; }
当时,我以为 cout << sth 和 cin >> xxx 这是“必须的格式”;这只是运算符重载在标准库里的一个缩影而已。这里实际调用的是<string>定义的:
extern template ostream& operator<<(ostream&, const char*);
容器与operator[]
下面展示vector和map因提供了operator[]而使程序变得简洁的实例。
vector::operator[]
STL 容器(Container)中的vector,map,都提供了operator[],对于vector,operator[]使得它的用法“和数组类似”,就是可以用下标访问vector的元素:
int firstInt = ivec[0]; // operator[] ivec[0] = 1; //
如果没有运算符重载,同样的功能很可能就要写成:
int firstInt = ivec.get(0); ivec.set(0, 1);
这就不再像数组那么“亲切”了。
下面的代码是求vector<int> ivec内所有元素和的代码:
int sum = 0; for(int i=0; i < ivec.size(); i++) { sum += ivec[i]; }
map::operator[]
类似的,operator[]使map非常好用。比如使用标准库map和string的单词统计的核心代码只有如下几行:string word; map<string, int> dict; while(cin >> word) { dict[word]++; // operator[] }
可以从cplusplus.com上看到,对于map,如果没有operator[],那上面的 dict[word]++ 一行要写成:
map<string, int>::iterator it = dict.find(word); if(it != dict.end()) { it->second++; } else { dict.insert(make_pair(word, 1)); }
使用C++标准库实现的"单词统计",整个程序如下:
#include <cstdio> #include <iostream> #include <map> #include <string> using namespace std; int main(int argc, char *argv[]) { string word; map<string, int> dict; while(cin >> word) { dict[word]++; } // output: for(map<string, int>::iterator it = dict.begin(); it != dict.end(); ++it) { cout << it->first << "\t" << it->second << "\n"; } return 0; }这段程序不仅完成了“单词统计”,还按照单词的字典顺序进行输出,这些全依赖于标准库的运算符重载。
迭代器与operator++
上面“单词统计”的代码,已经使用设计到了iterator,正是“迭代器”。简单地说,迭代器就是有指针功能的class类型;而它的“指针”功能,正是经由运算符重载实现的。
比如下面代码可以输出vecotr<int> ivec的全部元素:
for(vector<int>::iterator it = ivec.begin(); it != ivec.end(); // operator!= it++) { // operator++ printf("%d\n", *it); // operator* }这段短短的代码调用了iterator重载的三个operator。运算符重载使得这里for循环的写法和数组的迭代方式类似。
关于迭代器,最为激进的莫过于:
copy(istream_iterator<char>(cin), istream_iterator<char>(), ostream_iterator<char>(cout, ""));
string与operator+=
标准库的string,提供了operator[],使得用户可以使用下标运算符访问字符串中的字符,这和char array, char pointer无异。例如:
str1[0] = str2[0]; str2[0] = 'A';
除此之外,string还提供了重载了的operator+=,可以向已有的string对象追加字符和字符串(包括char array,char pointer)。例如:
str1 += '!'; str1 += str2; str1 += "literal string";
函数对象与operator()
在<algorithm>提供的众多算法中,大多都有两个版本,其中一个版本多出一个叫做Function Object的参数,比如sort:
template <class RandomAccessIterator> void sort ( RandomAccessIterator first, RandomAccessIterator last ); template <class RandomAccessIterator, class Compare> void sort ( RandomAccessIterator first, RandomAccessIterator last, Compare comp );comp就被称作是Function Object。
到底什么是Function Object呢?字面理解就是一个可以当函数调用的对象,其实就是一个重载了operator()的对象,比如要实现对一个vector<string>按照字符串长度对元素排序,可以传入一个这个Functor的实例:
struct StrLenComp { bool operator()(const string& a, const string& b) { return a.length() < b.length(); } };当然,如果你对C++11很熟悉,这个Functor的Function Object完全可以用一行的lambda表达式表示:
[](const string& a, const string& b){ return a.length() < b.length(); }
小结
上面列出的是标准库中最广为认知的运算符重载的例子,但标准库使用运算符重载的地方远不止此。
实质
C++中运算符重载实际上和成员函数并没有两样,只是写起来更简洁一点罢了。编译时,它们都会被修改为编译器内部的名称,也同样支持“重载”——参数列表不同。
C++运算符重载的妙用