216. 组合总和 III

找出所有相加之和为 n 的 k 个数的组合。组合中只允许含有 1 - 9 的正整数,并且每种组合中不存在重复的数字。

说明:

所有数字都是正整数。
解集不能包含重复的组合。 
示例 1:

输入: k = 3, n = 7
输出: [[1,2,4]]
示例 2:

输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/combination-sum-iii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

import java.util.ArrayList;
import java.util.Collections;
import java.util.LinkedList;
import java.util.List;

class Solution {

    private List<List<Integer>> ans = new ArrayList<>();

    private LinkedList<Integer> path = new LinkedList<>();

    private void solve(int k, int n, int pre) {
        if (k == 0 || n == 0) {
            if (n == 0 && k == 0) {
                ans.add(new ArrayList<>(path));
            }
            return;
        }
        for (int i = pre + 1; i <= 9; ++i) {
            if (i == n || (n - i > i)) {
                path.offerLast(i);
                solve(k - 1, n - i, i);
                path.pollLast();
            }
        }
    }

    public List<List<Integer>> combinationSum3(int k, int n) {
        if (n == 0) {
            return Collections.emptyList();
        }
        solve(k, n, 0);
        return ans;
    }
}
posted @ 2021-12-28 10:02  Tianyiya  阅读(34)  评论(0)    收藏  举报