如何将N个整数读入向量?

How to read N integers into a vector?

本文关键字:向量 整数      更新时间:2023-10-16

如果我想从标准输入读取所有整数到向量,我可以使用方便的:

vector<int> v{istream_iterator<int>(cin), istream_iterator()};

但是,假设我只想读取n整数。手动键入的循环是我得到的一切吗?

vector<int> v(n);
for(vector<int>::size_type i = 0; i < n; i++)
cin >> v[i];

还是有没有更右手的方法可以做到这一点?

通常不应该使用std::copy_n执行此操作,它假设提供的迭代器在递增n次时仍然有效:

count值从first开始的范围精确复制到从result开始的范围。形式上,对于每个非负整数i < n,执行*(result + i) = *(first + i)

(cppreference.com 关于std::copy_n的文章)

如果你能保证这一点,那就好了,但一般来说,std::cin这是不可能的。你可以很容易地让它取消引用一个无效的迭代器:

默认构造的std::istream_iterator称为流结束迭代器。当有效std::istream_iterator到达基础流的末尾时,它将等于流结束迭代器。取消引用或递增它会进一步调用未定义的行为。

(cppreference.com 关于std::istream_iterator的文章)

你的循环几乎在那里,尽管我可能会使用更强的终止条件来避免从"死"流中读取过多:

vector<int> v(n);
for(vector<int>::size_type i = 0; i < n; i++)
if (!cin >> v[i])
break;

我实际上很想把它包装成类似std::copy_n的东西,但接受一个完整的"范围",除了从0N计数之外,还可以验证其边界。

实现可能如下所示:

template<class InputIt, class Size, class OutputIt>
OutputIt copy_atmost_n(InputIt first, InputIt last, Size count, OutputIt result)
{
for (Size i = 0; i < count && first != last; ++i)
*result++ = *first++;
return result;
}

你可以像这样使用它:

copy_atmost_n(
std::istream_iterator<int>(std::cin),
std::istream_iterator<int>(),
N,
std::back_inserter(v)
);

现在你得到 M 个元素,其中M是提供的输入数或N,以较小者为准。

(现场演示)

如注释中所述,copy_n对于这项工作来说是不安全的,但您可以将copy_if与可变的 lambda 一起使用:

#include <iterator>
#include <vector>
#include <iostream>
#include <algorithm>
int main(){
const int N = 10;
std::vector<int> v;
//optionally v.reserve(N);
std::copy_if(
std::istream_iterator<int>(std::cin),
std::istream_iterator<int>(), 
std::back_inserter(v), 
[count=N] (int)  mutable {
return count && count--;
});
return 0;
}

正如这个答案所指出的: std::复制n个元素或到最后