累加和为 K 的子数组问题作者:Grey
原文地址:
博客园:累加和为 K 的子数组问题
CSDN:累加和为 K 的子数组问题
题目说明数组全为正数,且每个数各不相同,求累加和为K的子数组组合有哪些,
注:数组中同一个数字可以无限制重复被选取。如果至少一个数字的被选数量不同,则两种组合是不同的 。
题目链接见:LeetCode 39. Combination Sum
主要思路使用动态规划来解,定义如下递归函数
List<List<Integer>> p(int[] arr, int len, int i, int k)
递归含义表示:数组从 i 开始,一直到最后,可以得到的子数组满足数组之和等于 k 的子数组组合有哪些 。
首先是 base case
if (i == len) {return new ArrayList<>();}List<List<Integer>> ans = new ArrayList<>();if (k == 0) {ans.add(new ArrayList<>());return ans;}
当 i 到数组结尾位置下一个位置,说明,i 到头了,不能继续往后找了,直接返回一个空列表,
当 k 等于 0,直接返回一个包含空列表的列表,表示一个数也没有,组合之和等于 0 。
接下来是普遍情况,枚举每个位置有 times 个的情况下,往后收集的集合数是多少,即
for (int times = 0; times * arr[i] <= k; times++) {// 每个位置有 times 的情况下,往后收集的集合个数}
由于数组中全是正数,所以前提是: times * arr[i] <= k
,
如果times * arr[i]
正好等于 k,说明收集到了一个满足条件的集合,这个集合里面有 times 个 arr[i]
,
for (int times = 0; times * arr[i] <= k; times++) {// 每个位置有 times 的情况下,往后收集的集合个数if (times * arr[i] == k) {List<Integer> t = new ArrayList<>();// 收集到了一种情况,即集合里面有 times 个 arr[i]for (int j = 0; j < times; j++) {t.add(arr[i]);}ans.add(t);return ans;}……}
接下来就是当前位置 i 搞定 times * arr[i]
,i + 1 以后的数组搞定k - times * arr[i]
,
for (int times = 0; times * arr[i] <= k; times++) {if (times * arr[i] == k) {List<Integer> t = new ArrayList<>();for (int j = 0; j < times; j++) {t.add(arr[i]);}ans.add(t);return ans;}// 剩下的位置搞定 k - arr[i] * timesfor (List<Integer> a : p(arr, len, i + 1, k - times * arr[i])) {for (int j = 0; j < times; j++) {a.add(arr[i]);}ans.add(a);}}return ans;
完整代码如下
class Solution {// 从i号位置开始及其后面的所有,帮我搞定targetpublic static List<List<Integer>> combinationSum(int[] arr, int k) {return p(arr, arr.length, 0, k);}// 从i号位置开始及其后面的所有,帮我搞定targetprivate static List<List<Integer>> p(int[] arr, int len, int i, int k) {if (i == len) {return new ArrayList<>();}List<List<Integer>> ans = new ArrayList<>();if (k == 0) {ans.add(new ArrayList<>());return ans;}for (int times = 0; times * arr[i] <= k; times++) {if (times * arr[i] == k) {List<Integer> t = new ArrayList<>();for (int j = 0; j < times; j++) {t.add(arr[i]);}ans.add(t);return ans;}for (List<Integer> a : p(arr, len, i + 1, k - times * arr[i])) {for (int j = 0; j < times; j++) {a.add(arr[i]);}ans.add(a);}}return ans;}}
更多算法和数据结构笔记
【累加和为 K 的子数组问题】
经验总结扩展阅读
- 辛金男和戊土女的关系 五行相生女命旺夫
- 壬水命的女人漂亮 天性敏感为人直爽
- 壬水男喜欢什么女命 丁火为上等婚配
- 壬水为什么宠爱乙木 两命之人性格互补
- 属虎和属兔人感情走势颇高 爱情甜蜜十分登对
- 有和纯阳八字的人结婚吗 性格刚强事业好
- 五行缺水男和五行缺火女 性格互补似朋友
- 壬水男和辛金婚配吗 子孙昌盛家庭和睦
- 五行缺水女和五行缺金男在一起 改变关系更和睦
- 癸水男最佳婚配 戌土甲木庚辰均为良缘