leetcode 40 组合总和 II C语言

题目

leetcode 40
给定一个数组 candidates 和一个目标数 target ，找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的每个数字在每个组合中只能使用一次。

说明：
所有数字（包括目标数）都是正整数。
解集不能包含重复的组合。

示例 1:
输入: candidates = [10,1,2,7,6,1,5], target = 8,
所求解集为:
[
[1, 7],
[1, 2, 5],
[2, 6],
[1, 1, 6]
]

示例 2:
输入: candidates = [2,5,2,1,2], target = 5,
所求解集为:
[
[1,2,2],
[5]
]

代码

回溯法实现，注意去重，详见代码

int cmp(const void *a, const void *b) {
    return *(int *)a - *(int *)b;
}
void dfs(int *candidates, int ind, int candidatesSize, int target, int nums[], int n, int **arr, int *returnSize, int **returnColumnSizes) {
    if (target < 0) return ;
    if (target == 0) {
        arr[*returnSize] = (int *)malloc(sizeof(int) * n);
        for (int i = 0; i < n; i++) {
            arr[(*returnSize)][i] = nums[i];
        }
        (*returnColumnSizes)[*returnSize] = n;
        (*returnSize)++;
        return ;
    }
    int i = ind;
    //注意去重
    while (i < candidatesSize) {
        nums[n] = candidates[i];
        dfs(candidates, i + 1, candidatesSize, target - nums[n], nums, n + 1, arr, returnSize, returnColumnSizes);
        while (i < candidatesSize - 1 && candidates[i] == candidates[i + 1]) i++;
        i++;
    }
}
int** combinationSum2(int* candidates, int candidatesSize, int target, int* returnSize, int** returnColumnSizes){
    *returnSize = 0;
    if (candidatesSize == 0) return 0;
    qsort(candidates, candidatesSize, sizeof(int), cmp);
    int **arr = (int **)malloc(sizeof(int *) * 1000000);
    (*returnColumnSizes) = (int *)malloc(sizeof(int) * 1000000);
    int nums[target];
    dfs(candidates, 0, candidatesSize, target, nums, 0, arr, returnSize, returnColumnSizes);
    return arr;
}