[Golang]力扣Leetcode - 441. 排列硬币（累加、二分）

[Golang]力扣Leetcode - 441. 排列硬币（累加、二分）

题目：你总共有 n 枚硬币，并计划将它们按阶梯状排列。对于一个由 k 行组成的阶梯，其第 i 行必须正好有 i 枚硬币。阶梯的最后一行 可能 是不完整的。

给你一个数字 n ，计算并返回可形成 完整阶梯行 的总行数。

链接： 力扣Leetcode - 441. 排列硬币.

示例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))
}
 

提交截图：