题解 | #牛棚分组#
牛棚分组
https://www.nowcoder.com/practice/d5740e4dde3740828491236c53737af4
知识点:回溯,全排列
这是一道经典的数组全排列问题,只不过我们只需要对k个数进行排列,我们使用回溯的方法向数组中添加元素,当列表中元素达到k个时,就将其保存下来作为一种排列情况,否则,就继续向列表中添加元素,直至遍历完所有的情况。
Java题解如下
import java.util.*; public class Solution { /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param n int整型 * @param k int整型 * @return int整型二维数组 */ private List<List<Integer>> res = new ArrayList<>(); private int n; private int k; public int[][] combine (int n, int k) { // write code here this.n = n; this.k = k; backTracking(1, new ArrayList<>()); int[][] ans = new int[res.size()][k]; for(int i = 0; i < res.size(); i++) { for(int j = 0; j < k; j++) { ans[i][j] = res.get(i).get(j); } } return ans; } private void backTracking(int index, List<Integer> list) { for(int i = index; i <= n; i++) { list.add(i); if(list.size() == k) { res.add(new ArrayList<>(list)); }else { backTracking(i + 1, list); } list.remove(list.size() - 1); } } }