题解 | #牛棚分组# 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 背包问题 分治算法:归并排序、快速排序