今天帮人做蓝桥杯题目,正好需要用到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的值,进而影响其他循环
}
}
}
}
版权声明:本文为博主原创文章,未经博主允许不得转载。