[Golang]力扣Leetcode - 441. 排列硬币(累加、二分)
题目:你总共有 n 枚硬币,并计划将它们按阶梯状排列。对于一个由 k 行组成的阶梯,其第 i 行必须正好有 i 枚硬币。阶梯的最后一行 可能 是不完整的。
给你一个数字 n ,计算并返回可形成 完整阶梯行 的总行数。
示例1:
输入:n = 5
输出:2
解释:因为第三行不完整,所以返回 2 。
示例2:
输入:n = 8
输出:3
解释:因为第四行不完整,所以返回 3 。
思路:
-
法一:无脑累加
-
法二:二分法:定义查找的范围 [left,right],每次取查找范围的中点 mid,比较到 第 mid 层时硬币总和 和 总硬币 n 的大小,如果相等则 right 即为要寻找的层数,如果不相等则根据 第 mid 层时硬币总和 和 总硬币 n 大小关系将查找范围缩小。
- 如果 第 mid 层时硬币总和 <= 总硬币 n ,则 left = mid+1;
- 如果 第 mid 层时硬币总和 > 总硬币 n ,则 right = mid-1。
无脑累加Go代码:
package main
import "fmt"
func arrangeCoins(n int) int {
var sum, number int
for sum <= n {
number += 1
sum += number
}
return number - 1
}
func main() {
fmt.Println(arrangeCoins(5))
}
提交截图:
二分法Go代码:
package main
import "fmt"
func arrangeCoins(n int) int {
left,right := 1, n
for left<=right{
mid := (left+right)/2
if (1+mid)*mid/2 <= n{
left = mid+1
}else{
right = mid-1
}
}
return right
}
func main() {
fmt.Println(arrangeCoins(5))
}
提交截图: