我有这个功能:
void permutate (const string & s, std::vector<int>& index, std::size_t depth, int & count)
{
if (depth == s.size())
{
++count;
for (std::size_t i = 0; i < s.size(); ++i)
{
cout << s[index[i]];
}
cout << "\n";
return;
}
for (std::size_t i = 0; i < s.size(); ++i)
{
index[depth] = i;
permutate(s, index, depth + 1, count);
}
}
要打印所有组合,请使用:
void print(string s) {
if (s.find_first_not_of(s.front()) == string::npos)
{
cout << "Only 1 combination exists";
return;
}
sort(s.begin(), s.end());
cout << s << "\n**********\n";
vector<int> index(s.size());
int count = 0;
permutate(s, index, 0, count);
cout << "\nTotal combinations with repetitions: " << count;
}
功能很好但我需要特定的组合。例如,如果我写
打印(&#34; ABC&#34);
我得到:AAA,AAB,AAC,ABA,......,CCA,CCB,CCC。所以,我总共获得了27个组合。但是如果我需要组合大小不是原始组合(在这种情况下,原始组的大小是3)呢?
例如,如果我有一个字符串&#34; ABC&#34; (或一组S = {A,B,C}),我想要大小为2的组合,我应该得到9个组合:AA,AB,AC,BA,BB,BC,CA,CB,CC。
请建议为实现目标而进行的更改。谢谢。
答案 0 :(得分:0)
我认为最好的解决方案是输入新代码,并使用回溯进行排列。 我键入了类似的东西,你应该为你想要组合的字符提供向量V,并像bactrack(0,length_of_expected_string)一样调用函数bactrack。 您将按字母顺序在集合S中获得排列。 您可以在屏幕上取消注册typenig组合的代码。
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
#include<string>
#include<stack>
#include<list>
#include<vector>
#include<math.h>
#include<iomanip>
#include<fstream>
#include<cstdlib>
#include<ctime>
#include<map>
#include<set>
using namespace std;
vector<char> V;
vector<char> combination;
set<vector<char> > S;
void backtrack(int depth,int expected_length)
{
if(depth<expected_length)
{
for(int i=0;i<V.size();i++)
{
combination.push_back(V[i]);
backtrack(depth+1,expected_length);
combination.pop_back();
}
}
else
{
/*for(int i=0;i<combination.size();i++)
{
cout<<combination[i];
}
cout<<endl;*/
S.insert(combination);
}
}
int main()
{
backtrack(0,3);
for(auto it=S.begin();it!=S.end();it++)
{
/*
for(int i=0;i<(*it).size();i++)
{
cout<<(*it)[i];
}
cout<<endl;*/
}
}