如何为std::vector重载>>运算符并处理空输入终止循环？

原代码的问题 #

原代码中使用while(in >> elem)循环读取元素，由于标准输入的operator>>会自动跳过所有空白字符（包括换行、空格、制表符），因此用户单独按下回车键输入的空行会被直接忽略，循环会一直等待有效输入，无法终止。

解决方案 #

改用逐行读取+行内解析的逻辑，通过捕获空行来终止循环，同时保持对任意可输入类型T的支持。需要引入<sstream>头文件来使用std::istringstream解析每行内容：

#include <iostream>
#include <vector>
#include <sstream>
#include <string>

template<typename T>
std::istream& operator>>(std::istream& in, std::vector<T>& v)
{
    std::string line;
    while (std::getline(in, line))
    {
        // 检测到空行，终止输入循环
        if (line.empty())
        {
            break;
        }

        // 将当前行内容转为字符串流，解析其中的T类型元素
        std::istringstream iss(line);
        T elem;
        while (iss >> elem)
        {
            v.push_back(elem);
        }

        // 如果行内解析未到末尾（即存在无法转为T的内容），标记输入流错误
        if (!iss.eof())
        {
            in.setstate(std::ios::failbit);
            break;
        }
    }
    return in;
}

逻辑说明 #

1. 逐行读取：用std::getline读取整行输入，直接捕获用户输入的空行。
2. 空行判断：如果读取到的行是空字符串，立即终止循环，完成输入。
3. 行内解析：将非空行的内容存入std::istringstream，再通过标准的operator>>解析该行内的所有T类型元素，逐个添加到vector中。
4. 错误处理：如果行内存在无法转换为T的内容（比如输入了字母但T是int），会设置输入流的failbit，符合C++输入操作的错误处理规范。

使用示例 #

比如测试int类型的vector：

int main()
{
    std::vector<int> nums;
    std::cout << "输入整数，空行结束：\n";
    std::cin >> nums;

    std::cout << "输入的元素：";
    for (int num : nums)
    {
        std::cout << num << " ";
    }
    return 0;
}

输入：

1 2 3
4 5

程序会读取前两行的整数，第三行空行终止输入，输出输入的元素：1 2 3 4 5 。

内容的提问来源于stack exchange，提问作者ProMike