c++ - 使用 istream_iterator 并从标准输入或文件中读取

Question

我正在使用 Microsoft Visual C++ 编写程序，我希望我的程序使用 istream_iterator 从标准输入或文件中读取。谷歌搜索互联网并没有显示我认为它必须多么简单。因此，例如，我可以很容易地编写并从标准输入中读取:

#include <iostream>
#include <string>
#include <iterator>

using namespace std;

int main()
{
   istream_iterator<string> my_it(cin);
   for (; my_it != istream_iterator<string>(); my_it++)
      printf("%s\n", (*my_it).c_str());
}

或者我可以写这个并从文件中读取:

#include <iostream>
#include <string>
#include <iterator>
#include <fstream>

using namespace std;

int main(int argc, char** argv)
{
   ifstream file(argv[1]);
   istream_iterator<string> my_it(file);
   for (; my_it != istream_iterator<string>(); my_it++)
      printf("%s\n", (*my_it).c_str());
}

但我如何将这两者结合起来，以便一个简单的 (argc == 2) 检查让我用文件流或标准输入初始化我的输入流迭代器，然后继续我的快乐方式？

Answer 1

可以在构造迭代器后赋值给它:

int main(int argc, char** argv)
{
   ifstream file;
   istream_iterator<string> my_it;

   if(argc == 2) {
      file.open(argv[1]);
      my_it = istream_iterator<string>(file);
   } else {
      my_it = istream_iterator<string>(cin);
   }
}