[leetcode][216. 组合总和 III]

最新推荐文章于 2025-03-22 17:35:45 发布

原创最新推荐文章于 2025-03-22 17:35:45 发布 · 217 阅读

0 ·

CC 4.0 BY-SA版权

c++算法专栏收录该内容

19 篇文章

订阅专栏

本文深入探讨了寻找所有相加之和为n的k个数的组合算法，限制条件为组合中只允许含有1-9的正整数，且每种组合中不存在重复的数字。通过递归函数实现，确保解集不包含重复的组合，提供了具体示例并附带源代码。

找出所有相加之和为 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]]

class Solution {
public:
    vector<vector<int>> result;
    void func(vector<int> &data, int start, int pos, int size, int tmpsum, int sum)
    {
        if(pos<start) return;
        
        do
        {
            data[pos]++;
            tmpsum++;
        }while(pos>0 && data[pos]<=data[pos-1]);
        if(data[pos]>9)
        {
            tmpsum-=data[pos];
            data[pos]=0;
            func(data,start,pos-1,size,tmpsum,sum);
        }
        else
        {      
            if(tmpsum>sum)
            {
                tmpsum-=data[pos];
                data[pos]=0;
                func(data,start,pos-1,size,tmpsum,sum);
            }
            else if(tmpsum==sum)
            {
                if(pos-start+1==size)
                {
                    //output
                    result.push_back(data);
                    
                    //next
                    tmpsum-=data[pos];
                    data[pos]=0;
                    func(data,start,pos-1,size,tmpsum,sum);
                }
                else
                {
                    func(data,start,pos+1,size,tmpsum,sum);  
                }            
            }
            else
            {
                if(pos-start+1==size)
                {
                    func(data,start,pos,size,tmpsum,sum);
                }
                else
                {
                    func(data,start,pos+1,size,tmpsum,sum);
                }        
            }
        }
    }
    vector<vector<int>> combinationSum3(int k, int n) {
        vector<int> data(k,0);
        func(data, 0, 0, k, 0, n);
        return result;
    }
};