使用std :: istream_iterator限制std :: copy的范围

时间:2011-05-10 17:00:12

标签: c++ stl iterator istream-iterator

我构建了一个最小的工作示例,以显示我在使用STL迭代器时遇到的问题。我正在使用istream_iterator来阅读floats中的std::istream个(或其他类型):

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

int main() {
   float values[4];
   std::copy(std::istream_iterator<float>(std::cin), std::istream_iterator<float>(), values);
   std::cout << "Read exactly 4 floats" << std::endl; // Not true!
}

这将读取所有可能的floats,直到EOF变为values,其大小固定为4,所以现在显然我想限制范围以避免溢出并准确读取/最多4值。

使用更多“普通”迭代器(即RandomAccessIterator),如果begin+4没有超过您的结束:

std::copy(begin, begin+4, out);

准确阅读4个元素。

如何使用std::istream_iterator执行此操作?显而易见的想法是将对std::copy的调用更改为:

std::copy(std::istream_iterator<float>(std::cin), std::istream_iterator<float>(std::cin)+4, values);

但(相当可以预见)这不会编译,operator+没有候选人:

g++ -Wall -Wextra test.cc
test.cc: In function ‘int main()’:
test.cc:7: error: no match for ‘operator+’ in ‘std::istream_iterator<float, char, std::char_traits<char>, long int>(((std::basic_istream<char, std::char_traits<char> >&)(& std::cin))) + 4’

有什么建议吗?是否有正确的“STLified”前C ++ 0x方法来实现这一目标?显然我可能只是把它写成for循环,但我想在这里学习一些关于STL的东西。我想知道滥用std::transformstd::merge等以某种方式实现此功能,但我不知道该怎么做。

3 个答案:

答案 0 :(得分:14)

当您请求非C ++ 0x解决方案时,这里有一个替代方法,它使用std::generate_n和生成器函子而不是std::copy_n和迭代器:

#include <algorithm>
#include <string>
#include <istream>
#include <ostream>
#include <iostream>

template<
    typename ResultT,
    typename CharT = char,
    typename CharTraitsT = std::char_traits<CharT>
>
struct input_generator
{
    typedef ResultT result_type;

    explicit input_generator(std::basic_istream<CharT, CharTraitsT>& input)
      : input_(&input)
    { }

    ResultT operator ()() const
    {
        // value-initialize so primitives like float
        // have a defined value if extraction fails
        ResultT v((ResultT()));
        *input_ >> v;
        return v;
    }

private:
    std::basic_istream<CharT, CharTraitsT>* input_;
};

template<typename ResultT, typename CharT, typename CharTraitsT>
inline input_generator<ResultT, CharT, CharTraitsT> make_input_generator(
    std::basic_istream<CharT, CharTraitsT>& input
)
{
    return input_generator<ResultT, CharT, CharTraitsT>(input);
}

int main()
{
    float values[4];
    std::generate_n(values, 4, make_input_generator<float>(std::cin));
    std::cout << "Read exactly 4 floats" << std::endl;
}

如果您愿意,可以将此生成器与boost::generator_iterator结合使用,以将生成器用作输入迭代器。

答案 1 :(得分:13)

查看std::copy_n

答案 2 :(得分:3)

如果您没有std::copy_n,那么编写自己的代码非常简单:

namespace std_ext { 
template<class InputIterator, class Size, class OutputIterator>
OutputIterator copy_n(InputIterator first, Size n, OutputIterator result) {
    for (int i=0; i<n; i++) {
        *result = *first;
        ++result;
        ++first;
    }
    return result;
}
}