[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))
}

提交截图:
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/a6661314/article/details/124676581