Leetcode77. 组合

Every day a leetcode

题目来源：77. 组合

解法1：回溯算法

详情请见77.组合（中等）：回溯问题 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().
 */
#define MAX_ANSSIZE 10000
//定义一个全局变量路径（用于组合出组合）
int *path;
int pathSize;
//定义一个二维数组用于存放已经排好的一维数组
int **ans;
int ansSize;
//cur为当前元素
void dfs(int cur,int n,int k)
{
    //减枝
    if(pathSize+(n-cur+1)<k) return;

    if(pathSize == k)
    {
        //如果元素已经足够多了，则可以进行《放入操作》把该组合放入ans
        int*temp=malloc(k*sizeof(int));
        for(int i=0;i<k;i++) temp[i]=path[i];
        ans[ansSize++]=temp;
        return;
    }
    //对于没有满足上述条件的，说明组合可以组合出来但是元素不够多
    path[pathSize++]=cur;
    //把cur放入path
    dfs(cur+1,n,k);
    //然后进行回溯（把cur+1)放入dfs（cur+1在本循环中直到path满了或者元素不足时返回），(同时开启了cur+1的循环)
    pathSize--;
    //path去除组合中最后的一个元素（例 1.2中去除2可以加上3），该语句直会在返回满了之后才会用到
    dfs(cur+1,n,k);
    //再放入下一个元素进行上述所述的（1.3)组合，该语句不会开启循环，它所接到的path都是建立在pathSize--；的语句后面的，也就是组合中差一个元素的时候才会用到该语句
}

int** combine(int n, int k, int* returnSize, int** returnColumnSizes){
    path=malloc(k*sizeof(int));
    ans=malloc(MAX_ANSSIZE*sizeof(int*));
    pathSize=ansSize=0;
    dfs(1,n,k);
    //启动回溯
    *returnSize=ansSize;
    *returnColumnSizes=malloc(ansSize*sizeof(int*));
    for(int i=0;i<ansSize;i++)
    {
        (*returnColumnSizes)[i]=k;
    }
    return ans;
}

结果：