foreach據說已經進了新的C++標准,不過在沒有編譯器支持以前,自己寫一個也很容易。
(1)foreach 標准用法:
std::vector<int> vec;
foreach(int i, vec){
std::cout << i;
}
(2)VC實現
在最新的VC版本中原來已經有了類似於foreach的支持,改個名字就行了:
#define foreach(var, container)for each(var in containter)
(3)GCC實現
GCC沒有內嵌支持,不過由於GCC支持typeof關鍵字, 所以實現起來也不是太難. (有個bug, 在OwnWaterloo提醒下已經糾正)
template <typename C> struct foreach_helper {
typename C::const_iterator it, end;
foreach_helper (const C& c): it(c.begin()), end(c.end()){}
};
#define RANDOM_VAR(name, line)RANDOM_VAR_(name, line)
#define RANDOM_VAR_(name, line)name ## line
#define foreach(var, container) \
__typeof__(container)const& RANDOM_VAR(_con_, __LINE__)= container;\
for (foreach_helper <__typeof__(container)> _fh_(RANDOM_VAR(_con_, __LINE__)); _fh_.it != _fh_.end; ++_fh_.it)\
for (var = *_fh_.it;; __extension__ ({break;}))
這裡有一個特殊的考慮,就是container有可能是一個臨時對象,或者是某個函數的返回值。為了不對容器進行復制,利用了一個不太為人所知的C++特性,就是臨時變量在存在引用時,生命期會由引用變量決定。這樣保證在進行循環時始終有效。
(4)性能
我分別使用GCC和VC9進行了測試(優化選項都使用O2),結果表明使用foreach和普通的iterator 遍歷幾乎沒有差別。不過gcc的遍歷性能要明顯好於VC9 (用個具有中國特色的結論,就是大約要好五倍),我的測試當然很粗略,不值得相信。