题目是这样的:
求从n个不同元素中取r个(1<r<=n)元素与另外m个相同元素组成的排列。
假设n个不同元素是1到n,m个相同元素为m个0。
其中用到的算法在之前的博客里面有写到,此处直接给出代码。
import java.util.HashSet;
import java.util.Iterator;
import java.util.Scanner;
import java.util.Stack;
public class Permutation
{
public static void main(String[] args)
{
int n,r,m;
Scanner scanner =new Scanner(System.in);
System.out.println("please input n : ");//题目中给定的n
n=scanner.nextInt();
System.out.println("input r : ");//从n中选择r个元素
r=scanner.nextInt();
System.out.println("input m : ");//给定m个零
m=scanner.nextInt();
int arrayn[]=new int[n];//初始化n个数据
for(int i=0;i<n;i++)
{
arrayn[i]=i+1;
}
HashSet<String>results=new HashSet<String>();//用来存放排列之后得到的序列
Stack<Integer>stack=new Stack<Integer>();//用来求指定r个元素的组合
zuhe(arrayn,0,r,stack,m,results);
Iterator<String>it=results.iterator();
while(it.hasNext())
{
System.out.println((String)it.next());
}
}
public static void zuhe( int array[],int begin,int num,Stack<Integer> stack,int num1,HashSet<String>hashSet)
{//该方法采用递归的思路,求array中num个数的组合
if(num==0)
{
int lastArray[]=new int[stack.size()+num1];
Iterator<Integer> it=stack.iterator();
for(int i=0;i<stack.size()+num1;i++)
{
if(i<stack.size())lastArray[i]=(Integer) it.next();
else lastArray[i]=0;
}
pailie(lastArray,0,lastArray.length,hashSet);
}
else if(begin==array.length)
return;
else
{
stack.push(array[begin]);
zuhe(array,begin+1,num-1,stack,num1,hashSet);
stack.pop();
zuhe(array,begin+1,num,stack,num1,hashSet);
}
}
public static void pailie(int array[],int start ,int length,HashSet<String>hashSet)
{//该方法用来求array的元素的全排列
if(start==length)
{
String result="";
for(int i=0;i<array.length;i++)
{
result+=array[i];
}
hashSet.add(result);
}
else
{
for(int i=start;i<length;i++)
{
int temp=array[start];
array[start]=array[i];
array[i]=temp;
pailie(array,start+1,array.length,hashSet);
temp=array[start];
array[start]=array[i];
array[i]=temp;
}
}
}
}