前言

今天看到一个超级简单的算法题，但是我当时思路往递归，逐级筛选里面想了。结果百度查查答案，超级简单。
真是惭愧惭愧，不过我还是坚持用递归实现了，因为用递归的方案，可以适用于任何给定数据和指定位数。

传统解法

如下所示，因为题目是找1、2、3、4组合的三位数，因此可以用三重循环，遍历所有组合，筛选不重复组合即可。
但是该方案，如果给定数据改变，组合位数改变，那代码就得大改，所以不是一个通用的好方法。

func useNormal() []int {var ret []intfor i := 1; i < 5; i++ {for j := 1; j < 5; j++ {for k := 1; k < 5; k++ {if i != j && j != k && i != k {ret = append(ret, i*100+j*10+k)}}}}return ret
}

递归求解

下面方案是通过递归，依次为某一位分配数字，并将剩余组合依次赋值下一位。
该方法只需要改变data数据，以及numLen的指定位数就可以灵活地计算所有组合。

func useRecursion() []int {var (data   = []int{1, 2, 3, 4}numLen = 3tmp    = make([]int, numLen)ret    []int)findNum(data, tmp, numLen, &ret)return ret
}func findNum(data, tmp []int, dep int, ret *[]int) {for i, v := range data {if dep--; dep < 0 {sum := 0for i := len(tmp) - 1; i >= 0; i-- {sum = sum*10 + tmp[i]}*ret = append(*ret, sum)return}tmp[dep] = v // 当前选择赋值next := make([]int, 0, len(data))next = append(next, data[:i]...)next = append(next, data[i+1:]...)findNum(next, tmp, dep, ret) // 剔除当前元素,进入下一级筛选dep++}
}

完整代码

执行结果如下，两种方案结果完全一致：

[123 124 132 134 142 143 213 214 231 234 241 243 312 314 321 324 341 342 412 413 421 423 431 432] 24[123 124 132 134 142 143 213 214 231 234 241 243 312 314 321 324 341 342 412 413 421 423 431 432] 24

package mainimport "fmt"func main() {ret0 := useNormal()fmt.Println(ret0, len(ret0))ret1 := useRecursion()fmt.Println(ret1, len(ret1))
}func useNormal() []int {var ret []intfor i := 1; i < 5; i++ {for j := 1; j < 5; j++ {for k := 1; k < 5; k++ {if i != j && j != k && i != k {ret = append(ret, i*100+j*10+k)}}}}return ret
}func useRecursion() []int {var (data   = []int{1, 2, 3, 4}numLen = 3tmp    = make([]int, numLen)ret    []int)findNum(data, tmp, numLen, &ret)return ret
}func findNum(data, tmp []int, dep int, ret *[]int) {for i, v := range data {if dep--; dep < 0 {sum := 0for i := len(tmp) - 1; i >= 0; i-- {sum = sum*10 + tmp[i]}*ret = append(*ret, sum)return}tmp[dep] = v // 当前选择赋值next := make([]int, 0, len(data))next = append(next, data[:i]...)next = append(next, data[i+1:]...)findNum(next, tmp, dep, ret) // 剔除当前元素,进入下一级筛选dep++}
}

总结

凡事有简单方案，也有复杂方案，简单往往不能通用，多想想通用方案，尝试解决一类问题而不是某一个问题。