题意
输入一组数字(可能包含重复数字),输出其所有的排列方式
样例
输入:[1,2,3]
输出:
[
[1,2,3],
[1,3,2],
[2,1,3],
[2,3,1],
[3,1,2],
[3,2,1]
]
注意这里数组中的元素会出现相同的情况,如果是不相同的数算法简单一下方案数是(n!)
这道题目的思路是采用枚举的方法把数字填入坑中,但是如何处理数字相同的情况?方法是先通过sort排序把数字相同的放在一起如果前后两个数字相同那么第二个数字放的坑位必须在第一个之后如果不相同可以从头开始放
class Solution {
public:
vector<vector<int>> res;
vector<int> path;
int n;
void dfs(vector<int> &num,int u,int start,int state)//u表示num数组的枚举,start 表示相同数字上一次放入坑的位置 state表示状态1表示这里面有数字0表示这里面没有数字
{
if(u==n)
{
res.push_back(path);
return;
}
if(!u||num[u]!=num[u-1]) start=0;//前后的数字不相同则放入的坑位可以从0开始
for(int i=start;i<n;i++)//枚举要填入几号坑所以path的下标是i注意u是指将nums数组中的索引
{
if(!(state>>i&1))
{
path[i]=num[u];
dfs(num,u+1,i+1,state+(1<<i));//state+(1<<i)表示将状态集合里面的第i位置成1;
}
}
}
vector<vector<int>> permutation(vector<int>& nums) {
n=nums.size();
path.resize(n);
sort(nums.begin(),nums.end());
dfs(nums,0,0,0);
return res;
}
};
另一道经典的枚举题
最短Hamilton路径
题意
给定一张 n 个点的带权无向图,点从 0~n-1 标号,求起点 0 到终点 n-1 的最短Hamilton路径。 Hamilton路径的定义是从 0 到 n-1 不重不漏地经过每个点恰好一次。
输入格式
第一行输入整数n。
接下来n行每行n个整数,其中第i行第j个整数表示点i到j的距离(记为a[i,j])。
对于任意的x,y,z,数据保证 a[x,x]=0,a[x,y]=a[y,x] 并且 a[x,y]+a[y,z]>=a[x,z]。
样例
输出格式
输出一个整数,表示最短Hamilton路径的长度。
数据范围
1≤n≤20
0≤a[i,j]≤107
输入样例:
5
0 2 4 5 1
2 0 6 5 3
4 6 0 8 3
5 5 8 0 5
1 3 3 5 0
输出样例:
18
#include <iostream>
#include <cstring>
using namespace std;
const int N=20,M=1<<20;
int f[M][N];
int weight[N][N];
int main()
{
int n;
cin>>n;
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
{
cin>>weight[i][j];
}
memset(f,0x3f,sizeof(f));//将f全部变成正无穷;
f[1][0]=0;//表示一个状态在第0个地点不许要任何花费 i代表方案的集合1代表选择经过这个点0代表不选择这个点 j代表当前现在所在的点
for(int i=0;i<1<<n;i++)//i代表着是一个方案集合,其中每一个位置1和0,代表着这个点经过还是没有经过
for(int j=0;j<n;j++)//枚举当前到了哪个点
{
if(i>>j&1)//枚举出如果与1选择过这个点
for(int k=0;k<n;k++)//枚举到达j的点k
if((i-(1<<j))>>k&1)
f[i][j]=min(f[i][j],f[(i-(1<<j))][k]+weight[k][j]);
}
cout<<f[(1<<n)-1][n-1]<<endl;
return 0;
}
总结枚举的难点是应该按怎样的顺序进行枚举