C++ コマンドラインの引数が増加した場合

Question

C++の勉強中です。
下記のようにコマンドラインから引数としてA,B,Cのいずれの組み合わせを入力
すると、表示されるように作成しました。

このコードを変更したいと思っており、現状A,B,Cの組み合わせは
全部で15種類なのですが、組み合わせのパターンが増えることを想定した
サンプルコードなど教えていただけますでしょうか？
（例）組み合わせがA,B,C,D,E,Fなど

#define などで文字を追加し組み合わせのパターンを増やしたり、減らしたり対応できる
ようにしたいです。
よろしくお願い致します。


#include <iostream>
#include <string>
#include <unordered_set>
using namespace std;

int main(int argc, char* argv[])
{
    string str;
    static const unordered_set<string> ptn
    {
        "A",
        "A,B",
        "A,C",
        "A,B,C",
        "A,C,B",
        "B",
        "B,A",
        "B,C",
        "B,A,C",
        "B,C,A",
        "C",
        "C,A",
        "C,B",
        "C,A,B",
        "C,B,A",
    };
    bool found = false;
    int cnt = 1;

    if(ptn.count(argv[cnt]))
    {
        found = true;
    }
    if(found)
    {
        cout << "コマンドラインの引き数は : " << argv[cnt] << endl;
    }
    else
    {
        cout << "error: 入力誤りです"<< endl;
    }
}

Comments

[takasima20]

ソートしたのと比較するならそんなに多くならないんじゃ?

Answer 1

こんなのはいかがでしょうか？

#include <iostream>
#include <sstream>   // istringstream
using namespace std;

const int n = 3;  // n <= 26.  3 for A,B,C.  5 for A,B,C,D,E.

bool check(const char *s)
{
	istringstream iss(s);
	int cnt[26] = { 0 };
	char c;
	while (iss >> c)
		if (c < 'A' || c >= 'A' + n || cnt[c - 'A']++ ||
			iss >> c && c != ',') return false;
	return true;
}

int main(int argc, char* argv[])
{
	bool found = false;
	if (argc == 2) 
		found = check(argv[1]);
    if (found)
        cout << "コマンドラインの引き数は : " << argv[1] << endl;
    else
        cout << "error: 入力誤りです"<< endl;
}

追記
全パターンを生成したいのなら、

#include <iostream>
using namespace std;

const int n = 3;  // n <= 26.  3 for A,B,C.  5 for A,B,C,D,E.
char a[n], used[n];

void show(int k)
{
	for (int i = 0; i < k; i++) cout << a[i] << ',';
	cout << a[k] << '\n';
}

void generate(int i)
{
	if (i == n) return;
	for (int j = 0; j < n; j++)
		if (!used[j])
			used[j] = a[i] = 'A'+j, show(i), generate(i+1), used[j] = 0;
}

int main() { generate(0); }

追記2
最初のコードは、A,B,C, のように最後が , の場合を
入力謝りにしていなかったので、次のように訂正します。

#include <iostream>
#include <sstream>   // istringstream
using namespace std;

const int n = 3;  // n <= 26.  3 for A,B,C.  5 for A,B,C,D,E.

bool check(const char *s)
{
    istringstream iss(s);
    int cnt[26] = { 0 };
    char c;
    while (iss >> c && c >= 'A' && c < 'A'+n && !cnt[c-'A']++) {
		if (!(iss >> c)) return true;
        if (c != ',') break;
	}
	return false;
}

int main(int argc, char* argv[])
{
    bool found = false;
    if (argc == 2) 
        found = check(argv[1]);
    if (found)
        cout << "コマンドラインの引き数は : " << argv[1] << endl;
    else
        cout << "error: 入力誤りです"<< endl;
}

Comments

[episteme]

- 指定された文字以外出現せず
- 文字の重複がない
を満足すればOKならこれでいいのか。
# あたしゃ愚直に満足するパターンを列挙したけど

[episteme]

> 追記
まいりましたー orz

Answer 2

これでいいんじゃないかなー...

#include <iostream>
#include <string>
#include <algorithm>
#include <numeric>
#include <array>
#include <vector>

int main() {
  using namespace std;

  const int N = 5;

  // base = "ABCDE..." の反転
  string base(N,' ');
  iota(base.begin(), base.end(), 'A');
  reverse(base.begin(), base.end());

  vector<string> combination;
  for ( int M = 1; M <= 3; ++M ) {

    // bits : N-M個のfalse と M個の true
    array<bool,N> bits;
    fill(bits.begin(), bits.end(), false);
    fill(bits.begin()+N-M, bits.end(), true);

    // bitsに基づいたpatternを生成し、combinationに挿入
    std::string pattern;
    do {
      pattern.clear();
      for ( int i = 0; i < N; ++i ) {
        if ( bits[i] ) pattern += base[i]; 
      }
      reverse(pattern.begin(),pattern.end());
      combination.emplace_back(pattern);
    } while ( next_permutation(bits.begin(), bits.end()) );
  }

  // combination内のpatternそれぞれに対し順列を生成
  for ( string pattern : combination ) {
    do {
      string result;
      for ( char ch : pattern ) {
        if ( !result.empty() ) result += ',';
        result += ch;
      }
      cout << result << endl;
    } while ( next_permutation(pattern.begin(),pattern.end()) );
  }
}