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';
}
}