思路:回溯三部曲:
- 定义两个全局变量,result存放结果集,path存放任何条件的结果。 startIndex 记录下一层递归搜索的起始位置。
- 终止条件:当到达叶子节点,即pathTop == k时,result收集path,return。
- 单层搜索过程。处理节点、递归函数、回溯操作。
关于剪枝操作:画图可以观察出,要完成剪枝操作,只需在单层搜索过程中改动即可。
搜索起点的上界 = n-还需要的元素个数 +1; 即 j<= n (k-pathTop) + 1;
int *path;
int pathTop;
int **result;
int resultTop;
void backTrack(int n,int k,int startIndex){
if(pathTop == k){
int *temp = malloc(sizeof(int)*k);
for(int i=0;i<k;i++){
temp[i] = path[i];
}
result[resultTop++] = temp;
return ;
}
//注释的是没有剪枝的操作
//for(int j=startIndex; j<=n ; j++){
for(int j=startIndex; j<= n-(k-pathTop)+1 ; j++){
path[pathTop++] = j;
backTrack(n,k,j+1);
pathTop --;
}
}
int** combine(int n, int k, int* returnSize, int** returnColumnSizes){
pathTop = 0;
resultTop = 0;
path = malloc(sizeof(int)*k);
result = malloc(sizeof(int*) *10000);
backTrack(n,k,1);
*returnSize = resultTop ;
*returnColumnSizes = malloc(sizeof(int)* (*returnSize));
for(int i = 0 ;i<(*returnSize);i++){
(*returnColumnSizes)[i] = k;
}
return result ;
}