题解 | #加起来和为目标值的组合(三)#
加起来和为目标值的组合(三)
https://www.nowcoder.com/practice/8c297161a58740c5b92b3e184dd1756e
import java.util.*;
public class Solution {
private List<List<Integer>> res = new ArrayList<>();
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param k int整型
* @param n int整型
* @return int整型二维数组
*/
public int[][] combination (int k, int n) {
// write code here
if (n < 1 || k < 1) {
return new int[0][0];
}
backtrack(k, 1, n, new ArrayList<>());
int[][] matrix = new int[res.size()][k];
for (int i = 0; i < matrix.length; i++) {
List<Integer> list = res.get(i);
for (int j = 0; j < list.size(); j++) {
matrix[i][j] = list.get(j);
}
}
return matrix;
}
private void backtrack(int k, int start, int target, List<Integer> list) {
if (list.size() == k && target == 0) {
res.add(new ArrayList<>(list));
return;
}
for (int i = start; i <= 9; i++) {
if (target - i < 0) {
break;
}
target -= i;
list.add(i);
backtrack(k, i + 1, target, list);
target += i;
list.remove(list.size() - 1);
}
}
}
