C++實現輸入多行數字到數組

前天做某公司筆試題的時候，其輸入格式是多行數字，每行以空格為分隔符，以換行符號為結束輸入到多個數組。在JAVA中有相應的函數直接將一行拆成數組，感覺在C++中這中輸入方式還是挺奇怪的，今天想出一種解決方式。

思路：

每次讀入一個字符，判斷是不是EOF，如果是的話跳出循環；

不是EOF將字符放回緩沖區；

讀入換行符，說明一行結束，處理數組；

讀入字符，如果不是空格，存入一個臨時的字符串；

如果是空格，字符串轉換成整數並壓入數組；

注意：

對於連續的空格輸入要進行判斷，否則會輸入一堆0；

在讀入換行符的時候要將最後的臨時字符串存進數組；

代碼：

#include<iostream>
#include<string>
#include<vector>
#include<cstdlib>
using namespace std;

int main() {
    char flag;
    while ((flag=getchar())!=EOF) {
        putchar(flag);
        string tmpStr;
        vector<int> buff;
        char c;
        while ((c = getchar()) != '\n') {
            if (c != ' ')
                tmpStr.push_back(c);
            else {
                if (tmpStr != "") {
                    buff.push_back(atoi(tmpStr.c_str()));
                    tmpStr = "";
                }
            }
        }
        if(tmpStr!="")
            buff.push_back(atoi(tmpStr.c_str()));
        for (auto a : buff)
            cout << a << ' ';
        cout << '\n';
    }
}