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