C++算法入门练习——子集、全排列、组合问题

给定一个长度为n的序列，其中有n个可能重复的正整数，求该序列的所有子集、全排列、组合数。 

子集样例：

全排列样例：

组合样例：

对于这三种看起来不一样的问题，但实际上他们的套路都是DFS递归。

并且组合其实就是子集的特例情况。（从子集取满足要求的集合）

先说子集的解题思路：

其实就是利用两个vector，一个存结果子集，一个存临时取的数。

递归边界是：当层数到达了n+1时候，把vector temp中的数据给vector result，同时这里因为输入的数字中有重复的，因此需要去重，那么就要利用到一个set，我们去计算这个集合组成的数是否出现过，出现过就不存入result，否则存入，然后把该数存入set。

正常就是先取一个num数组里一个数存进temp，然后递归下一层。到递归边界后，将最上头的数弹出，然后继续递归一次。

同时因为输出有格式要求，我们要写一个cmp函数，利用algorithm里的sort快排一遍result再输出。

代码如下：

#include <iostream>
#include <vector>
#include <set>
#include <algorithm>
#include <cmath>
using namespace std;

vector<vector<int> > result;
vector<int> temp;
int n;
int num[13] = {0};

set<int> shuchu;

void DFS(int idx) {
    if (idx == n + 1) {
    	int numtemp = 0;
    	for(int i=0;i<temp.size();i++){
    		numtemp += temp[i]*pow(10,i);
            //cout<<numtemp<<endl;
		}
    	if(shuchu.find(numtemp)==shuchu.end()){
			result.push_back(temp);
			shuchu.insert(numtemp);
			return ;	
		}
        else{
            return ;
        }
    }
    temp.push_back(num[idx]);
    DFS(idx + 1);
    temp.pop_back();
    DFS(idx + 1);
}

bool cmp(vector<int> &a, vector<int> &b) {
    if (a.size() != b.size()) {
        return a.size() < b.size();
    } else {
        return a < b;
    }
}

int main() {
    scanf("%d", &n);
    for(int i=1;i<=n;i++){
    	cin>>num[i];
	}
    DFS(1);
    sort(result.begin(), result.end(), cmp);
    for (int i = 0; i < result.size(); i++) {
        for (int j = 0; j < result[i].size(); j++) {
            printf("%d", result[i][j]);
            if (j + 1 < result[i].size()) {
                printf(" ");
            }
        }
        printf("\n");
    }
    return 0;
}

而组合，只需要在递归边界里再加入一条语句if（temp.size()==k)，即temp里包含的元素是要组合数要取的个数，才存入result。

代码如下：

#include <iostream>
#include <vector>
#include <algorithm>
#include <set>
#include <cmath>
using namespace std; 

int n;
int k;
vector<int> temp;
vector<vector<int>> result;
set<int> shuchu;
int num[20];

void dfs(int idx){
	if(idx==n+1){
        int numtemp = 0;
		if(temp.size()==k){
            for(int i=0;i<temp.size();i++){
                numtemp += temp[i]*pow(10,i);
            }
            if(shuchu.find(numtemp)==shuchu.end()){
				result.push_back(temp);
				shuchu.insert(numtemp);
			}
        }
		return;
	}
	temp.push_back(num[idx]);
	dfs(idx+1);
	temp.pop_back();
	dfs(idx+1);
}

bool cmp(vector<int> &a,vector<int> &b){
	return a<b;
}


int main(){
	cin>>n;
	cin>>k;
	for(int i=1;i<=n;i++){
		cin>>num[i];
	}
	dfs(1);
	sort(result.begin(),result.end(),cmp);
	for (int i = 0; i < result.size(); i++) {
        for (int j = 0; j < result[i].size(); j++) {
            printf("%d", result[i][j]);
            if (j + 1 < result[i].size()) {
                printf(" ");
            }
        }
        printf("\n");
    }
	return 0;
}

而全排列问题，跟子集问题不一样，因为子集问题中，我们没有调换原本数字存放的顺序。

因此，我们要调整dfs的结构。 

正确做法应该是：增加一个状态变量，检测当前位数是否已经有数字，有了就不给，没有就给。

结束一次dfs后要将状态置回false。其中递归过程推荐画一下递归图，就可以理解了。

代码如下：

#include <iostream>
#include <set>
#include <cmath>
#include <vector>
using namespace std;

int n;
int numset[20];
vector<vector<int>> result;
vector<int> vtemp;
bool numstate[101] = {false};	
set<int> shuchu;

void DFS(int i){
	if(i==n+1){
		int temp = 0;
		for(int j=1;j<=n;j++){
			temp += vtemp[j-1]*pow(10,j-1);
		}
		if(shuchu.find(temp)==shuchu.end()){
			result.push_back(vtemp);
			shuchu.insert(temp);
			return ;	
		}
	}
	for(int j=1;j<=n;j++){
		if(!numstate[j]){
			numstate[j] = true;
			vtemp.push_back(numset[j]);
			DFS(i+1);
			numstate[j] = false;
			vtemp.pop_back();
		}
	}
}

int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>numset[i];
	}
	DFS(1);
	for(int j=0;j<result.size();j++){
		for(int i=0;i<n;i++){
			if(i!=n-1){
				cout<<result[j][i]<<" ";
			}
		else{
			cout<<result[j][i]<<endl;
			}	
		}
	}
	return 0;
}