题解 | #牛棚分组# Java

牛棚分组

https://www.nowcoder.com/practice/d5740e4dde3740828491236c53737af4

用list方便添加和删除(回溯)

最后再转成数组

import java.util.*;


public class Solution {
    public static ArrayList<ArrayList<Integer>> lists = new
    ArrayList<ArrayList<Integer>>();
    public int[][] combine (int n, int k) {
        backTracking(1, new ArrayList<>(), n, k);

        int[][] ans = new int[lists.size()][k];
        for (int i = 0; i < lists.size(); i++) {
            for (int j = 0; j < k; j++) {
                ans[i][j] = lists.get(i).get(j);
            }
        }

        return ans;
    }

    void backTracking(int index, List<Integer> list, int n, int k) {
        for (int i = index; i <= n; i++) {
            list.add(i);
            if (list.size() == k) {
                lists.add(new ArrayList<>(list));
            } else {
                backTracking(i + 1, list, n, k);
            }
            list.remove(list.size() - 1);
        }

    }
}
算法题刷刷刷 文章被收录于专栏

数组、链表、栈、队列、堆、树、图等。 查找和排序:二分查找、线性查找、快速排序、归并排序、堆排序等。 动态规划:背包问题、最长公共子序列、最短路径 贪心算法:活动选择、霍夫曼编码 图:深度优先搜索、广度优先搜索、拓扑排序、最短路径算法(如 Dijkstra、Floyd-Warshall) 字符串操作:KMP 算法、正则表达式匹配 回溯算法:八皇后问题、0-1 背包问题 分治算法:归并排序、快速排序

全部评论

相关推荐

醉蟀:你不干有的是人干
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务