问题描述
什么是组合,相信打开这篇博文的你应该是知道的,我在此不再赘述,为了方便描述,该问题简述为n个数中选m个进行组合,求所有组合。
解法一:for循环
如果n和m为给定的值,可以考虑该解法,为简化问题,n个数为1到n的数,假设是6选3进行组合
public class ZuHe {
public static void main(String[] args) {
int[] arr = {1,2,3,4,5,6};
int count = 0;
//6选3
for (int i=0;i<arr.length;i++){
for (int j=i+1;j<arr.length;j++){
for (int k=j+1;k<arr.length;k++){
count++;
System.out.println(arr[i]+","+arr[j]+","+arr[k]);
}
}
}
System.out.println("组合总数:"+count);
}
}
此种解法不够灵活,如果m为动态的值,可参考下面的解法。
解法二:递归回溯
采用递归的解法
public class ZuHe {
static List<List<Integer>> resList = new ArrayList<List<Integer>>();
static int[] arr = {1,2,3,4,5,6};
static int count = 0;
public static void main(String[] args) {
LinkedList<Integer> list = new LinkedList<Integer>();
dfs(1,list,3);//6选3进行组合
System.out.println("组合总数:"+count);
for (List<Integer> list1 :resList){
System.out.println(list1);
}
}
static void dfs(int i, LinkedList<Integer> list, int length){
if(i == length+1){
count++;
resList.add((List<Integer>) list.clone());
return;
}
for (int a:arr){
if (list.size() == 0){//选第一个数时随便选,没有限制
list.add(a);
dfs(i+1,list,length);
list.pollLast();
}else{
//选非第一个数时,选择的数都要比已选的所有数都大,这是为了避免重复组合
int b = list.get(list.size()-1);
if(a > b){
list.add(a);
dfs(i+1,list,length);
list.pollLast();
}
}
}
}
}
解法三:位运算
思路:位运算适合于求所有组合的情况。比如给定1,2,3三个数,求所有组合的情况。很明显有 1 ,2 ,3 ,1 2 ,1 3 ,2 3 ,1 2 3,七中组合方式 。按照2进制的思想的话,这七种组合分别对应1 0 0(第一个数选,第二三个数不选),0 1 0(第一三个数不选,第二个数选),0 0 1,1 1 0,1 0 1,0 1 1,1 1 1,其中1表示该位数选,0表示改位数不选,其实还有一个组合就是0 0 0表示都不选。细心的你可能已经发现了规律:n个数求组合,一共有 1 << n (2的n次方)种组合,所有组合情况为0到2的n次方的二进制表示形式,用0表示不选该位数,1表示选择该位数。遍历0到2的n次方的所有数,即可拿到所有组合情况。