题解 | #牛棚分组#

牛棚分组

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);
        }
    }
}

全部评论

相关推荐

点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务