用do_if和assign_if函数来提高一贯性

标准模板库（STL）为你提供了许多算法，如：for_each、copy、transform以及许多其它算法。你可能已经注意到，许多算法还有名字上添加词缀‘_if’的对应算法，如counte_if、remove_copy_if、replace_copy_if。不幸的是，其它的一些算法没有与之对应的算法，这就显得有些不一致了。

我们将向你介绍一种解决这个问题的方法。这个方法对任何满足下列条件的算法都适用：

• 仿函式（如for_each）
• 输出叠代器（output iterator，例如copy、copy_n、merge、rotate_copy、set_difference）

当一个算法需要仿函式时，你就可以使用do_if，当do_if中的一元判断返回为“真”时，算法要求的仿函式即被调用。Assign_if的用法与之类似，只有当assign_if中的一元判断返回为“真”时，才把元素值赋给输出叠代器。你可以用assign_if来筛选输出序列。

上面两个函数的用法很简单：

• do_if( Operation op,UnaryPredicate p)：对元素e,当判断条件p(e)返回为真时执行操作op(e)。
• assign_if(OutputIterator it, UnaryPredicate p)：对元素e，当判断条件p(e) 返回为真时，赋值给它。

它们的用法很简单，例如：

// 用for_each实现只打印小于10的元素 std::for_each( aTest, aTest + nCount, do_if( std::ptr_fun( print), is_less( 10))); // 只合并偶元素 std::merge( aFirst, aFirst + nFirstCount, aSecond, aSecond + nSecondCount, assign_if( std::back_inserter( aMerge), std::ptr_fun( is_even)));

下面是do_if和assign_if的代码和相应的例子：

#include <functional> template< class Operation, class UnaryPred> struct do_if_t { do_if_t( Operation operation, UnaryPred pred) : m_operation( operation), m_pred( pred) {} template< class Type> void operator()( Type & value) const { if ( m_pred( value)) m_operation( value); } private: Operation m_operation; UnaryPred m_pred; }; template< class OutputIterator, class UnaryPred> struct assign_if_t { assign_if_t( OutputIterator it, UnaryPred pred) : m_iterator( it), m_pred( pred), m_bHasSucceeded( true) {} template< class Type> void operator=( Type & value) { m_bHasSucceeded = m_pred( value); if ( m_bHasSucceeded) *m_iterator = value; } assign_if_t & operator*() { return *this; } assign_if_t & operator++() { increment(); return *this; } assign_if_t & operator++( int) { increment(); return *this; } assign_if_t & operator--() { decrement(); return *this; } assign_if_t & operator--( int) { decrement(); return *this; } operator OutputIterator () { return m_iterator; } private: void increment() { if ( m_bHasSucceeded) ++m_iterator; } void decrement() { if ( m_bHasSucceeded) --m_iterator; } private: OutputIterator m_iterator; UnaryPred m_pred; // has the last operation succeeded? bool m_bHasSucceeded; }; template< class Operation, class UnaryPred> inline do_if_t< Operation, UnaryPred> do_if( Operation op, UnaryPred p) { return do_if_t< Operation, UnaryPred>( op, p); } template< class OutputIterator, class UnaryPred> inline assign_if_t< OutputIterator, UnaryPred> assign_if( OutputIterator it, UnaryPred p) { return assign_if_t< OutputIterator, UnaryPred>( it, p); } //////////////////////////////////// //例子 #include <iostream> #include <algorithm> #include <vector> template< class type> inline std::binder2nd< std::less< type> > is_less( type value) { return std::bind2nd( std::less< type>(), value); } int print( int i) { std::cout << i << " "; return 0; } bool is_even( int i) { return ( i % 2) == 0; } int main(int argc, char* argv[]) { int aTest[] = { 3, 6, 1, 30, 4, 7, 20, 3, 34, 2, 66 }; int nCount = sizeof( aTest) / sizeof( int); // print only elements less than 10, using for_each std::for_each( aTest, aTest + nCount, do_if( std::ptr_fun( print), is_less( 10))); std::cout << std::endl; // print only elements less than 10, using std::copy std::copy( aTest, aTest + nCount, assign_if( std::ostream_iterator< int>( std::cout, " "), is_less( 10))); // 合并两个已排序的序列 int aFirst[] = { -10, -9, -6, -3, -1, 0, 4, 6, 7, 10 }; int nFirstCount = sizeof( aFirst) / sizeof( int); int aSecond[] = { -7, -5, -3, -2, 0, 3, 5, 6, 10 }; int nSecondCount = sizeof( aSecond) / sizeof( int); std::vector< int> aMerge; //合并所有的元素 std::merge( aFirst, aFirst + nFirstCount, aSecond, aSecond + nSecondCount, std::back_inserter( aMerge)); std::cout << " ALL Merged elements: " << std::endl; std::copy( aMerge.begin(), aMerge.end(), std::ostream_iterator< int>( std::cout, " ")); // 只合并偶元素 aMerge.clear(); std::merge( aFirst, aFirst + nFirstCount, aSecond, aSecond + nSecondCount, assign_if( std::back_inserter( aMerge), std::ptr_fun( is_even))); std::cout << " EVEN Merged elements: " << std::endl; std::copy( aMerge.begin(), aMerge.end(), std::ostream_iterator< int>( std::cout, " ")); return 0; }