算法训练营day23(补),回溯3

import (

"sort"

)

39. 组合总和

func combinationSum(candidates []int, target int) [][]int {

//存储全部集合

result := make([][]int, 0)

if len(candidates) == 0 {

return result

}

sort.Ints(candidates) //排序后面做剪枝

//存储单次集合

path := make([]int, 0)

var backtrace func(candidates []int, target int, startIndex int)

backtrace = func(candidates []int, target int, startIndex int) {

if target == 0 {

temp := make([]int, len(path))

copy(temp, path)

result = append(result, temp)

return

}

for i := startIndex; i < len(candidates); i++ {

if candidates[i] > target { //剪枝

break

}

path = append(path, candidates[i])

backtrace(candidates, target-candidates[i], i)

//回溯处理

path = path[:len(path)-1]

}

}

backtrace(candidates, target, 0)

return result

}

40. 组合总和 II

func combinationSum2(candidates []int, target int) [][]int {

//存储全部集合

result := make([][]int, 0)

if len(candidates) == 0 {

return result

}

sort.Ints(candidates) //排序后面做剪枝

//记录数组每一个元素是否使用过

user := make([]bool, len(candidates))

//存储单次集合

path := make([]int, 0)

var backtrace func(candidates []int, target int, startIndex int)

backtrace = func(candidates []int, target int, startIndex int) {

if target == 0 {

temp := make([]int, len(path))

copy(temp, path)

result = append(result, temp)

return

}

for i := startIndex; i < len(candidates); i++ {

if candidates[i] > target { //剪枝

break

}

if i > 0 && candidates[i] == candidates[i-1] && user[i-1] == false { //过滤重复

continue

}

path = append(path, candidates[i])

user[i] = true

backtrace(candidates, target-candidates[i], i+1)

//回溯处理

path = path[:len(path)-1]

user[i] = false

}

}

backtrace(candidates, target, 0)

return result

}

//判断是否是回文

func isPalindrome(s string) bool {

for i, j := 0, len(s)-1; i < j; i, j = i+1, j-1 {

if s[i] != s[j] {

return false

}

}

return true

}

131. 分割回文串

func partition(s string) [][]string {

//存储全部集合

result := make([][]string, 0)

if len(s) == 0 {

return result

}

//存储单次集合

path := make([]string, 0)

var backtrace func(se string, startIndex int)

backtrace = func(se string, startIndex int) {

if startIndex == len(se) {

temp := make([]string, len(path))

copy(temp, path)

result = append(result, temp)

return

}

for i := startIndex; i < len(se); i++ {

//对字符串进行切割

str := se[startIndex : i+1]

if isPalindrome(str) {

path = append(path, str)

backtrace(se, i+1)

//回溯处理

path = path[:len(path)-1]

}

}

}

backtrace(s, 0)

return result

}

相关推荐
程序员爱钓鱼28 分钟前
Go操作Word文档实战:github.com/nguyenthenguyen/docx
后端·google·go
火莲华1 小时前
go cond 探索
go
AI工程架构师1 小时前
通常说算力是多少 FLOPS,怎么理解,GPU和CPU为什么差异这么大
算法
祈安_1 小时前
Java实现循环队列、栈实现队列、队列实现栈
java·数据结构·算法
归去_来兮15 小时前
拉格朗日插值算法原理及简单示例
算法·数据分析·拉格朗日插值
千寻girling21 小时前
Python 是用来做 AI 人工智能 的 , 不适合开发 Web 网站 | 《Web框架》
人工智能·后端·算法
程序员爱钓鱼1 天前
GoHTML解析利器:github.com/PuerkitoBio/goquery实战指南
后端·google·go
颜酱1 天前
一步步实现字符串计算器:从「转整数」到「带括号与优化」
javascript·后端·算法
我叫黑大帅1 天前
Go中的interface的两大用法
后端·面试·go