遍历m个元素取k个的递归解法

今天帮人做蓝桥杯题目，正好需要用到m个元素取k个值，把其中的方法提炼出来作为工具类：

import java.util.Arrays;
import java.util.ArrayList;

/**
 * 遍历m个元素取k个。
 * 将问题抽象为在数组arr[m]中取k个元素，将所有取法存入ArrayList<int[]>中。
 */
public class Combination{
    private static ArrayList<int[]> list = new ArrayList<int[]>(); //用于储存所有的下标组合
    private static int[] tempArr; //用于存储每一次遍历的下标组合

    //输入需求
    public static void set(int m, int k){ //m个元素中取k个
        tempArr = new int[k];
        getNum(m-1, k);
    }

    //返回遍历结果
    public static ArrayList<int[]> get(){
        return list;
    }

    //不返回，直接输出遍历结果
    public static void print(){
        int[] temp;
        for(int i = 0; i <list.size(); i++){
            temp = list.get(i);
                System.out.println(Arrays.toString(temp));
        }
    }

    private static void getNum(int mMax, int k){ //mMax为待选取子数组的最大下标，k为须取的个数,取出存入tempArr[k]中
        if(k == 1) {
            for(int i = mMax ; i >= 0; i--){
                tempArr[k-1] = i; //既然是最后一位，下标不需再自增
                int[] copy = Arrays.copyOf(tempArr, tempArr.length);
                list.add(copy);//直接add原数组会导致所有list元素都指向原数组
            } 
        }
        else{ 
            for(int i = mMax; i >= k-1; i--){                
                tempArr[k-1] = i;
                getNum(i-1, k-1); //这里如果用自增符号，会直接改变i,k的值，进而影响其他循环
            }
        }
    }

}

版权声明：本文为博主原创文章，未经博主允许不得转载。