Question

我想用以下代码中的算法替换循环

int numbers[] = { ... };
vector<int> output;

for( int* it = numbers+from; it != numbers+to ; ++it ) 
{
    int square = func( *it );
    if( predicate(square) )
    {
         output.push_back(square);
    }
}

该程序旨在转换值并在条件发生时将它们复制到目的地。

我无法使用std::copy_if，因为这不会应用转换。
我无法使用std::transform因为缺少谓词
由于转换变量的中间副本，编写 transform_copy_if() 并不是一个好主意。

看起来我唯一的希望就是创建一个conditional_back_insert_iterator。然后我可以有一个相当不错的电话，如：

int numbers[] = { ... };
vector<int> output;

std::transform(numbers+from, numbers+to, 
               conditional_back_inserter(predicate, output),   
               func);

此解决方案是否是处理此类案件的最佳方法？我甚至无法使用条件插件器，因此我担心我走错了路。

我还可以想象我可以实施替代解决方案，例如

std::copy_if( transform_iterator<func>(numbers+from), 
              transform_iterator<func>(numbers+to), 
              back_inserter(output) );

（这让我想起了boost中的* filter_iterators *的一个例子）但这不提供可读性。

Answer 1

我认为创建自己的迭代器是可行的方法：

#include <iostream>
#include <vector>
#include <iterator>
#include <functional>

template<class T>
class conditional_back_insert_iterator
    : public std::back_insert_iterator<std::vector<T>>
{
private:
    using Base       = std::back_insert_iterator<std::vector<T>>;
    using Container  = std::vector<T>;
    using value_type = typename Container::value_type;
public:
    template<class F>
    conditional_back_insert_iterator(Container& other, F&& pred)
        :  Base(other), c(other), predicate(std::forward<F>(pred))
    { }

    conditional_back_insert_iterator<T>& operator*()
    { return *this; }

    conditional_back_insert_iterator<T>&
        operator=(const value_type& val) const
    {
        if (predicate(val))
            c.push_back(val);
        return *this;
    }

    conditional_back_insert_iterator<T>&
        operator=(value_type&& val) const
    {
        if (predicate(val))
            c.push_back(std::move(val));
        return *this;
    }
private:
    Container& c;
    std::function<bool (const value_type&)> predicate;
};

template<
    class Container,
    class F,
    class value_type = typename Container::value_type
>
conditional_back_insert_iterator<value_type>
    conditional_back_inserter(Container& c, F&& predicate)
{
    return conditional_back_insert_iterator<value_type>(c, std::forward<F>(predicate));
}

int main()
{
    std::vector<int> v{1, 2, 3, 4, 5, 6, 7, 8, 9};
    std::vector<int> to;

    auto is_even = [] (int x) { return (x % 2) == 0; };

    std::copy(v.begin(), v.end(), conditional_back_inserter(to, is_even));
}

Answer 2

这是我的尝试。

#include <algorithm>
#include <iostream>
#include <iterator>
#include <vector>

template <class Container, class Pred>
class conditional_insert_iterator 
    : public std::iterator< std::output_iterator_tag, void, void, void, void >
{
public:
    explicit conditional_insert_iterator(Container& c, Pred p) : container(&c), pred(p) {}
   conditional_insert_iterator& operator=(typename Container::const_reference value) {
    if (pred(value))
        container->push_back(value);
    return *this;
}

    conditional_insert_iterator& operator*() {return *this;}
    conditional_insert_iterator& operator++() {return *this;}
    conditional_insert_iterator& operator++(int) {return *this;}
private:
    Container* container;
    Pred pred;
};

template< class Container, class Pred>
conditional_insert_iterator<Container, Pred> conditional_inserter( Container& c, Pred pred )
{
    return conditional_insert_iterator<Container, Pred>(c, pred);
}

using namespace std;

int main()
{
    vector<int> in = { 1, 2, 3, 4, 5, 6 };
    vector<int> out;
    transform(in.begin(), in.end(),
              conditional_inserter(out, [](int i) { return i%2 == 0;}), 
              [](int i) { return i + 2;});
    for (auto i : out)
        cout << i << "\n"; 

    return 0;
}

模拟条件back_inserter之类的东西的最佳方法是什么？

2 个答案: