本文作者 TomorrowWu,原创文章,转载注明出处,博客地址 https://segmentfault.com/u/to... 第一时间看后续精彩文章。觉得好的话,顺手分享到朋友圈吧,感谢支持。
最近看了一道关于三色球的面试题,虽然很简单,但是还是考研编码能力的,下面做个对比
题目
有红、黄、绿三种颜色的球，其中红球3个，黄球3个，绿球6个。现将这12个球混放在一个盒子里，从中任意摸出8个球，编程计算摸出球的各种颜色搭配。
看到题目,显然是排列组合问题,下面看代码
优化前
func computeMatch1(redCount, yellowCount, greenCount, total int) int {
if redCount < 0 || yellowCount < 0 || greenCount < 0 || total <= 0 || total > redCount+yellowCount+greenCount {
return 0
}
num := 0
for r := 0; r <= redCount; r++ {
for y := 0; y <= yellowCount; y++ {
for g := 0; g <= greenCount; g++ {
if r+y+g == total {
//log.Printf("red:%v yellow:%v green:%v", r, y, g)
num++
}
}
}
}
return num
}测试结果
为精确测出算法计算的时间,注释掉了log.Printf()代码
func Test_computeMatch1(t *testing.T) {
t1 := time.Now()
n := computeMatch1(3000,3000,6001,12000)
duration := time.Since(t1)
t.Logf("排列组合数量:%v 时长:%v",n,duration)
}
//结果
排列组合数量:3 时长:32.450475506s代码未优化,三层for循环,且不管要求摸出多少球,遍历次数都一样,时间复杂度是O(n)=a*b*c,在球数比较大时,运算时间惊人
优化后
func computeMatch(redCount, yellowCount, greenCount, total int) int {
if redCount < 0 || yellowCount < 0 || greenCount < 0 || total <= 0 || total > redCount+yellowCount+greenCount {
return 0
}
num := 0
for r := 0; r <= redCount; r++ {
if yellowCount+greenCount < total-r {
continue
}
if r == total {
num++
//log.Printf("red:%v yellow:%v green:%v", r, 0, 0)
break
} else if r < total {
for y := 0; y <= yellowCount; y++ {
if r+y == total {
num++
//log.Printf("red:%v yellow:%v green:%v", r, y, 0)
break
} else if r+y < total {
needGreen := total - r - y
if needGreen > 0 && greenCount >= needGreen {
//log.Printf("red:%v yellow:%v green:%v", r, y, total-r-y)
num++
}
}
}
}
}
return num
}测试结果
func Test_computeMatch(t *testing.T) {
t1 := time.Now()
n := computeMatch(3000,3000,6001,12000)
duration := time.Since(t1)
t.Logf("排列组合数量:%v 时长:%v",n,duration)
}
//结果
排列组合数量:3 时长:106.502µs总结
- 第一点主要是解题思路,能迅速将问题对应到数学上的排列组合问题,初步思路就是三层for循环
- 第二点就是代码质量问题,因为三层for循环,循环次数累积相乘,当球数很大时,计算所消耗的时间很可怕,虽然只有3种组合方式,所以就针对题目特点,对第二层循环做了优化,第三层循环可以去掉,具体优化看代码,并不复杂
- 第三点就是对函数参数的边界检查,细心