在生成1~n的排列一文中,我们获取排列时用了控制不相等来实现让每个数都不重复地出现,所以如果要生成含有重复元素的全排列,对对生成1~n的排列的程序适当修改即可。
首先,把各个元素改成用户输入,输入数组为p,然后把if(a[k]==j)改为if(a[k]==p[j]),a[cur]=j改成a[cur]=p[j],然后先把p数组中元素用模板函数sort()按从小到大排个序,接下来要更改原程序中控制元素不重复出现的部分:
for(int k=0;k<cur;k++){
if(a[k]==p[j]){
ok++;break;
}
}
把它改为控制重复次数的代码:
int c1=0,c2=0;
for(int x=0;x<cur;x++)if(p[j]==a[x])c1++;
for(int y=0;y<n;y++)if(p[j]==p[y])c2++;
判断是否接收该元素也由条件
if(c1<c2)
决定。
下面是完整代码:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<sstream>
#include <cstdlib>
using namespace std;
void permutation(int *a,int *p,int n,int cur){
if(cur==n){
for(int i=0;i<n;i++)printf("%d ",a[i]);printf("\n");
}
else {
for(int j=0;j<n;j++)if(!j||p[j]!=p[j-1]){
int c1=0,c2=0;
for(int x=0;x<cur;x++)if(p[j]==a[x])c1++;
for(int y=0;y<n;y++)if(p[j]==p[y])c2++;
if(c1<c2){
a[cur]=p[j];
permutation(a,p,n,cur+1);
}
}
}
}
int main(){
int n;
scanf("%d",&n);
int *a=new int[n];
int *p=new int[n];
for(int i=0;i<n;i++)scanf("%d",&p[i]);
permutation(a,p,n,0);
delete(a);
delete(p);
}