每日一道算法题 组合

题目

77. 组合 - 力扣（LeetCode）

Python

体会一下Python的高开发效率

class Solution:
    def combine(self, n: int, k: int) -> List[List[int]]:
        from itertools import combinations
        return list(combinations(range(1,n+1),k))

第二种解法如下： 

 

class Solution:
    def combine(self, n: int, k: int) -> List[List[int]]:
        ans=[]
        sub_ans=[]
        def fun(supply):
            need=k-len(sub_ans)  # 需求，还需要几个数
            if need>supply: #需大于供
                return
            if need==0:
                ans.append(sub_ans.copy())
            for i in range(supply,0,-1):
                sub_ans.append(i)
                fun(i-1)
                sub_ans.pop()
        fun(n)
        return ans

C++

class Solution {
public:
    vector> combine(int n, int k) 
    {
        vector> ans;
        vector sub_ans;

        function fun=[&](int supply)
        {
            int need=k-sub_ans.size();
            if( need>supply) return;
            if(need==0)
            {
                ans.push_back(sub_ans);
                return;
            } 

            for(int i=supply;i>0;i--)
            {
                sub_ans.push_back(i);
                fun(i-1);
                sub_ans.pop_back();
            }
        };
        fun(n);
        return ans;
    }
};

第二种：

class Solution {
public:
    vector> combine(int n, int k)
    {
        vector> ans;
        vector sub_ans;

        function fun=[&](int num)
        {
            int need=k-sub_ans.size();
            if(need==0)
            {
                ans.push_back(sub_ans);
                return;
            }

            if(num>need) fun(num-1); //可以不选
            //选
            sub_ans.push_back(num);
            fun(num-1);
            sub_ans.pop_back();

        };

        fun(n);
        return ans;

    }
};

C语言

/**
 * Return an array of arrays of size *returnSize.
 * The sizes of the arrays are returned as *returnColumnSizes array.
 * Note: Both returned array and *columnSizes array must be malloced, assume caller calls free().
 */
int **ans;
int *sub_ans;
int idx_ans;
int idx_sub;
void fun(int n,int k,int startIdx)
{
    if(k==idx_sub)
    {
        //用tmp将sub_ans放到ans，直接使用sub_ans的话，值会变
        //不能指向sub_ans
        int *tmp=(int *)malloc(sizeof(int)*k);  //tmp[k]
        for(int i=0;i