当前位置:   article > 正文

回溯算法练习day.3

回溯算法练习day.3

39.组合总和

链接:. - 力扣(LeetCode)

题目描述:

给你一个 无重复元素 的整数数组 candidates 和一个目标整数 target ,找出 candidates 中可以使数字和为目标数 target 的 所有 不同组合 ,并以列表形式返回。你可以按 任意顺序 返回这些组合。

candidates 中的 同一个 数字可以 无限制重复被选取 。如果至少一个数字的被选数量不同,则两种组合是不同的。 

对于给定的输入,保证和为 target 的不同组合数少于 150 个。

示例 1:

输入:candidates = [2,3,6,7], target = 7
输出:[[2,2,3],[7]]
解释:
2 和 3 可以形成一组候选,2 + 2 + 3 = 7 。注意 2 可以使用多次。
7 也是一个候选, 7 = 7 。
仅有这两种组合。

提示:

  • 1 <= candidates.length <= 30
  • 2 <= candidates[i] <= 40
  • candidates 的所有元素 互不相同
  • 1 <= target <= 40

思路:

我们以集合[2,3,5]为例子,目标值为4,因为是组合问题,因此选择回溯算法来解决,因为可以选择重复的元素,因为我们可以抽象出如下的树形结构

回溯实现:

1.确定函数参数和返回值,回溯算法返回值一般为空,传入的应该是集合和目标值,统计当前的目标值和,开始遍历的位置

2.确定终止条件,当查找到我们的目标值时,记录值,如果当前值大于目标值则退出

3.确定单层递归逻辑,收集节点元素,更新当前值,递归,回溯

代码如下:

  1. int* path; // 用于存储当前组合的路径
  2. int pathTop; // 记录当前路径的长度
  3. int** result; // 存储所有和等于目标值的组合
  4. int resulttop; // 记录结果数组的长度
  5. // 记录每一个和等于target的path数组长度
  6. int* len; // 记录每个组合的长度
  7. void backTracking(int target, int index, int* candidates, int candidatesSize, int sum) {
  8. // 若sum大于等于target就应该终止遍历
  9. if(sum >= target) {
  10. // 若sum等于target,将当前的组合放入result数组中
  11. if(sum == target) {
  12. // 创建临时数组来存储当前路径
  13. int* tempPath = (int*)malloc(sizeof(int) * pathTop);
  14. int j;
  15. // 复制当前路径到临时数组中
  16. for(j = 0; j < pathTop; j++) {
  17. tempPath[j] = path[j];
  18. }
  19. // 将临时数组存入结果数组中
  20. result[resulttop] = tempPath;
  21. // 记录当前组合的长度
  22. len[resulttop++] = pathTop;
  23. }
  24. return ;
  25. }
  26. int i;
  27. for(i = index; i < candidatesSize; i++) {
  28. // 将当前数字加入sum
  29. sum += candidates[i];
  30. // 将当前数字加入路径
  31. path[pathTop++] = candidates[i];
  32. // 递归调用backTracking
  33. backTracking(target, i, candidates, candidatesSize, sum);
  34. // 回溯,将当前数字从sum和路径中移除
  35. sum -= candidates[i];
  36. pathTop--;
  37. }
  38. }
  39. int** combinationSum(int* candidates, int candidatesSize, int target, int* returnSize, int** returnColumnSizes){
  40. // 初始化变量
  41. path = (int*)malloc(sizeof(int) * 50);
  42. result = (int**)malloc(sizeof(int*) * 200);
  43. len = (int*)malloc(sizeof(int) * 200);
  44. resulttop = pathTop = 0;
  45. // 调用回溯函数找到所有组合
  46. backTracking(target, 0, candidates, candidatesSize, 0);
  47. // 设置返回的数组大小
  48. *returnSize = resulttop;
  49. *returnColumnSizes = (int*)malloc(sizeof(int) * resulttop);
  50. int i;
  51. for(i = 0; i < resulttop; i++) {
  52. // 将每个组合的长度存入返回数组中
  53. (*returnColumnSizes)[i] = len[i];
  54. }
  55. // 返回结果数组
  56. return result;
  57. }

 

40.组合总和II

链接:. - 力扣(LeetCode)

题目描述:

给定一个候选人编号的集合 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的每个数字在每个组合中只能使用 一次 。

注意:解集不能包含重复的组合。 

示例 1:

输入: candidates = [10,1,2,7,6,1,5], target = 8,
输出:
[
[1,1,6],
[1,2,5],
[1,7],
[2,6]
]

提示:

  • 1 <= candidates.length <= 100
  • 1 <= candidates[i] <= 50
  • 1 <= target <= 30

 思路:

因为题目给的数组里有重复元素,但是在解集中不能有重复的组合,因此我们要实现一个去重操作,我们可以使用一个数组来标记题目给的集合,当集合中的元素没有使用时,我们标记为0,如果使用过则标记为1,因为题目中要求不能使用同一个元素,这里的同一个元素是指下标位置相同的元素,而不是值相同,我们可以抽象为树形结构

在这里我们第一条路径下是从第1个1往后取的,它还会包括后面的路径,因为题目要求组合不能重复,因此后面的两个分支是没有必要的,如果再去搜索就会重复,因此这里去重的关键就是在每一层(横向)中进行去重,出现过的就不需要再去遍历,而纵向是不需要的,因为可以使用

回溯实现:

1.确定函数参数和返回值,回溯一般不需要返回值,参数应该为题目的集合,目标值,当前总和,每层开始遍历位置,使用一个数组标记我们使用过的元素

2.确定终止条件,如果当前值超过目标值,则退出,如果相等,则进行存储

3.确定单层递归,判断当前元素是否与前一个元素相同(排序后的集合)且是否它的标记数组的值为0,相同则进行去重

代码实现:

  1. /**
  2. * 返回一个大小为*returnSize的数组的数组。
  3. * 数组的大小作为*returnColumnSizes数组返回。
  4. * 注意:返回的数组和*columnSizes数组都必须是malloced,假设调用者调用free()。
  5. */
  6. int *path; // 路径数组
  7. int pathtop; // 路径数组的顶部索引
  8. int **result; // 结果数组
  9. int resulttop; // 结果数组的顶部索引
  10. int *len; // 每个结果数组的长度
  11. // 比较函数
  12. int cmp(const void *a1, const void *a2)
  13. {
  14. return *((int *)a1) - *((int *)a2);
  15. }
  16. // 回溯函数
  17. void backtracking(int *candidates, int candidatesSize, int target, int sum, int startindex)
  18. {
  19. // 如果当前累计和大于等于目标值
  20. if(sum >= target)
  21. {
  22. // 如果当前累计和等于目标值
  23. if(sum == target)
  24. {
  25. // 分配临时数组存储当前路径
  26. int *temp = malloc(sizeof(int) * pathtop);
  27. for(int i = 0; i < pathtop ; i++)
  28. temp[i] = path[i];
  29. // 将当前路径长度存储到结果数组的长度数组中
  30. len[resulttop] = pathtop;
  31. // 将当前路径存储到结果数组中
  32. result[resulttop++] = temp;
  33. }
  34. return ;
  35. }
  36. // 遍历候选数组
  37. for(int i = startindex; i < candidatesSize; i++)
  38. {
  39. // 如果当前元素与上一个相同,则跳过,避免重复组合
  40. if(i > startindex && candidates[i] == candidates[i-1])
  41. continue;
  42. // 将当前元素加入路径数组
  43. path[pathtop++] = candidates[i];
  44. // 更新累计和
  45. sum += candidates[i];
  46. // 递归调用回溯函数,更新累计和、路径数组的下一个起始索引
  47. backtracking(candidates, candidatesSize, target, sum, i+1);
  48. // 回溯,将当前元素从路径中移除
  49. sum -= candidates[i];
  50. pathtop--;
  51. }
  52. }
  53. int** combinationSum2(int* candidates, int candidatesSize, int target, int* returnSize, int** returnColumnSizes) {
  54. // 分配内存
  55. path = (int *)malloc(sizeof(int) * 50);
  56. result = (int **)malloc(sizeof(int *) * 100);
  57. len = (int *)malloc(sizeof(int) * 100);
  58. // 对候选数组排序
  59. qsort(candidates, candidatesSize, sizeof(int), cmp);
  60. pathtop = resulttop = 0;
  61. // 回溯
  62. backtracking(candidates,candidatesSize,target,0,0);
  63. // 设置返回的结果数量
  64. *returnSize = resulttop;
  65. // 分配每个结果数组的大小
  66. *returnColumnSizes = (int *)malloc(sizeof(int) * resulttop);
  67. for(int i = 0; i < resulttop; i++)
  68. (*returnColumnSizes)[i] = len[i];
  69. return result;
  70. }

注意:

qsort 函数是C标准库中的一个函数,用于对数组进行排序,其作用是对一个数组进行快速排序,可以按照用户提供的比较函数定义的规则,对数组元素进行排序,即qsort 函数的作用是将一个数组按照指定的排序规则进行重新排列,以使得数组中的元素按照指定的顺序排列,排序的顺序可以是升序或者降序,具体取决于用户提供的比较函数,原型如下:

void qsort(void *base, size_t nmemb, size_t size, int (*compar)(const void *, const void *));
  • base 是待排序数组的起始地址
  • nmemb 是数组中元素的个数
  • size 是每个元素的大小(以字节为单位)
  • compar 是一个函数指针,指向一个比较函数,用于确定元素的顺序

代码中的cmp函数是比较函数

它将 a1a2 强制转换为 int* 类型,然后分别取出其指向的整数值,并将其进行比较。函数的返回值是一个整数,表示两个值的大小关系:

如果 a1 所指向的值小于 a2 所指向的值,返回负数(<0)

如果 a1 所指向的值等于 a2 所指向的值,返回零(0)

如果 a1 所指向的值大于 a2 所指向的值,返回正数(>0)

这样定义的比较函数用于 qsort 函数时,可以根据返回值的不同,实现升序或降序排列

131.分割回文串

链接:. - 力扣(LeetCode)

题目描述:

给你一个字符串 s,请你将 s 分割成一些子串,使每个子串都是

回文串

。返回 s 所有可能的分割方案。

示例 1:

输入:s = "aab"
输出:[["a","a","b"],["aa","b"]]

示例 2:

输入:s = "a"
输出:[["a"]]

提示:

  • 1 <= s.length <= 16
  • s 仅由小写英文字母组成

思路:

因为是分割问题,因此可以使用回溯算法来解决,因此我们就可以抽象为一个树形结构

在这里就可以看出我们的结果都在叶子节点,就是已经将字符串完全切割之后所得的结果

函数实现:

1.确定参数和返回值,题目提供的字符串,每次开始切割的位置(相当于切割线),返回值一般为空

2.确定终止条件,当切割线到达末尾,则到达叶子节点,收集结果

3.确定单层递归,判断子串是否回文,如果是则收集子串,否则直接跳过,进行递归,回溯

代码如下:

  1. char** path; // 用于存储回文子串的数组
  2. int pathTop; // 记录path数组的当前索引
  3. char*** ans; // 存储最终结果的数组
  4. int ansTop = 0; // 记录ans数组的当前索引
  5. int* ansSize; // 记录每个结果的长度的数组
  6. void copy() {
  7. // 复制path数组到临时数组
  8. char** tempPath = (char**)malloc(sizeof(char*) * pathTop);
  9. int i;
  10. for(i = 0; i < pathTop; i++) {
  11. tempPath[i] = path[i];
  12. }
  13. // 将临时数组加入到结果数组中,并记录其长度
  14. ans[ansTop] = tempPath;
  15. ansSize[ansTop++] = pathTop;
  16. }
  17. bool isPalindrome(char* str, int startIndex, int endIndex) {
  18. // 检查给定范围内的字符串是否为回文
  19. while(endIndex >= startIndex) {
  20. if(str[endIndex--] != str[startIndex++])
  21. return 0;
  22. }
  23. return 1;
  24. }
  25. char* cutString(char* str, int startIndex, int endIndex) {
  26. // 截取给定范围内的字符串
  27. char* tempString = (char*)malloc(sizeof(char) * (endIndex - startIndex + 2));
  28. int i;
  29. int index = 0;
  30. for(i = startIndex; i <= endIndex; i++)
  31. tempString[index++] = str[i];
  32. tempString[index] = '\0';
  33. return tempString;
  34. }
  35. void backTracking(char* str, int strLen, int startIndex) {
  36. // 回溯函数,用于搜索所有可能的回文分割方案
  37. if(startIndex >= strLen) {
  38. // 如果已经遍历完字符串,将当前方案加入到结果中
  39. copy();
  40. return ;
  41. }
  42. int i;
  43. for(i = startIndex; i < strLen; i++) {
  44. if(isPalindrome(str, startIndex, i)) {
  45. // 如果从startIndex到i的子串是回文,则加入到path数组中
  46. path[pathTop++] = cutString(str, startIndex, i);
  47. }
  48. else {
  49. continue;
  50. }
  51. // 递归搜索剩余部分
  52. backTracking(str, strLen, i + 1);
  53. // 回溯,尝试其他可能的回文子串
  54. pathTop--;
  55. }
  56. }
  57. char*** partition(char* s, int* returnSize, int** returnColumnSizes){
  58. // 分割字符串s成回文子串的函数
  59. int strLen = strlen(s);
  60. path = (char**)malloc(sizeof(char*) * strLen);
  61. ans = (char***)malloc(sizeof(char**) * 40000);
  62. ansSize = (int*)malloc(sizeof(int) * 40000);
  63. ansTop = pathTop = 0;
  64. backTracking(s, strLen, 0);
  65. *returnSize = ansTop;
  66. *returnColumnSizes = (int*)malloc(sizeof(int) * ansTop);
  67. int i;
  68. for(i = 0; i < ansTop; ++i) {
  69. (*returnColumnSizes)[i] = ansSize[i];
  70. }
  71. return ans;
  72. }
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/从前慢现在也慢/article/detail/572725
推荐阅读
相关标签
  

闽ICP备14008679号