Need iterator when using ranged-based for loops

前端 未结 6 1963
失恋的感觉
失恋的感觉 2020-11-29 19:34

Currently, I can only do ranged based loops with this:

for (auto& value : values)

But sometimes I need an iterator to the value, instea

6条回答
  •  旧巷少年郎
    2020-11-29 20:04

    Here is a proxy wrapper class to allow you to expose the hidden iterator by aliasing it to your own variable.

    #include 
    #include 
    
    /*  Only provides the bare minimum to support range-based for loops.
        Since the internal iterator of a range-based for is inaccessible,
        there is no point in more functionality here. */
    template< typename iter >
    struct range_iterator_reference_wrapper
        : std::reference_wrapper< iter > {
        iter &operator++() { return ++ this->get(); }
        decltype( * std::declval< iter >() ) operator*() { return * this->get(); }
        range_iterator_reference_wrapper( iter &in )
            : std::reference_wrapper< iter >( in ) {}
        friend bool operator!= ( range_iterator_reference_wrapper const &l,
                                 range_iterator_reference_wrapper const &r )
            { return l.get() != r.get(); }
    };
    
    namespace unpolluted {
        /*  Cannot call unqualified free functions begin() and end() from 
            within a class with members begin() and end() without this hack. */
        template< typename u >
        auto b( u &c ) -> decltype( begin( c ) ) { return begin( c ); }
        template< typename u >
        auto e( u &c ) -> decltype( end( c ) ) { return end( c ); }
    }
    
    template< typename iter >
    struct range_proxy {
        range_proxy( iter &in_first, iter in_last )
            : first( in_first ), last( in_last ) {}
    
        template< typename T >
        range_proxy( iter &out_first, T &in_container )
            : first( out_first ),
            last( unpolluted::e( in_container ) ) {
            out_first = unpolluted::b( in_container );
        }
    
        range_iterator_reference_wrapper< iter > begin() const
            { return first; }
        range_iterator_reference_wrapper< iter > end()
            { return last; }
    
        iter &first;
        iter last;
    };
    
    template< typename iter >
    range_proxy< iter > visible_range( iter &in_first, iter in_last )
        { return range_proxy< iter >( in_first, in_last ); }
    
    template< typename iter, typename container >
    range_proxy< iter > visible_range( iter &first, container &in_container )
        { return range_proxy< iter >( first, in_container ); }
    

    Usage:

    #include 
    #include 
    std::vector< int > values{ 1, 3, 9 };
    
    int main() {
        // Either provide one iterator to see it through the whole container...
        std::vector< int >::iterator i;
        for ( auto &value : visible_range( i, values ) )
            std::cout << "# " << i - values.begin() << " = " << ++ value << '\n';
    
        // ... or two iterators to see the first incremented up to the second.
        auto j = values.begin(), end = values.end();
        for ( auto &value : visible_range( j, end ) )
            std::cout << "# " << j - values.begin() << " = " << ++ value << '\n';
    }
    

提交回复
热议问题