以下示例输出来自stdin的输入:

#include <iostream>

int main(int argc, char **argv)
{
    std::string s1, s2;
    while (getline(std::cin, s1))
    {
        std::cout << s1 << " 1\n";
    }

    while (getline(std::cin, s2))
    {
        std::cout << s2 << "2\n";
    }
    return 0;
}

我这样运行程序:cat main.cpp | ./main输出
#include <iostream> 1
 1
int main(int argc, char **argv) 1
{ 1
 1
    std::string s1, s2; 1
    while (getline(std::cin, s1)) 1
    { 1
        std::cout << s1 << " 1\n"; 1
    } 1
         1
    while (getline(std::cin, s2)) 1
    { 1
        std::cout << s2 << "2\n"; 1
    } 1
} 1

有没有办法通过stdin将两个单独的输入传递给程序?也就是说,如果我写类似cat main.cpp cat main.cpp | ./main的东西,那么第一个输入将分配给s1,第二个输入将分配给s2

最佳答案

AFAIK这是不可能的,但是您可以简单地使用包含输入的两个文件,将它们作为命令行参数传递并从中读取。

#include <iostream>
#include <fstream>

int main(int argc, char **argv)
{
    if (argc != 3)
      return -1;

    std::ifstream ifs1(argv[1]);
    std::ifstream ifs2(argv[2]);

    if (!ifs1.is_open() || !ifs2.is_open())
      return -2;

    std::string s1, s2;
    while (getline(ifs1, s1))
    {
      std::cout << s1 << " 1\n";
    }

    while (getline(ifs2, s2))
    {
      std::cout << s2 << "2\n";
    }
    return 0;
}

并这样称呼它:
./main main.cpp main.cpp

您甚至可以使其更加灵活,从而允许您读取所需数量的文件:
#include <iostream>
#include <fstream>

int main(int argc, char **argv)
{
    if (argc == 1)
      return -1; // no input file given

    for (int i = 1; i < argc; ++i) {
      std::ifstream ifs(argv[i]);

      if (!ifs.is_open()) {
        std::cout << argv[i] << ":file not found\n";
        continue;
      }

      std::string line;
      while (getline(ifs, line))
      {
        std::cout << line << " " << i << "\n";
      }
    }
}

07-24 09:45
查看更多