今天和大家聊的问题叫做 排列硬币,我们先来看题面:https://leetcode-cn.com/problems/arranging-coins/
You have n coins and you want to build a staircase with these coins. The staircase consists of k rows where the ith row has exactly i coins. The last row of the staircase may be incomplete.
Given the integer n, return the number of complete rows of the staircase you will build.
你总共有 n 枚硬币,并计划将它们按阶梯状排列。对于一个由 k 行组成的阶梯,其第 i 行必须正好有 i 枚硬币。阶梯的最后一行 可能 是不完整的。给你一个数字 n ,计算并返回可形成 完整阶梯行 的总行数。
示例
示例 1:
输入:n = 5输出:2解释:因为第三行不完整,所以返回 2 。
示例 2:
输入:n = 8输出:3解释:因为第四行不完整,所以返回 3 。
解题
用到等差数列求和公式。用二分法来做,就是寻找最后一个等差数列之和小于等于n 的数,也就是找右边界。
class Solution { public: int arrangeCoins(int n) { int l = 1, r = n; if (n == 0) return 0; while (l < r) { long long mid = l + (r - l + 1) / 2; if ((1 + mid) * mid / 2 <= n) l = mid; else r = mid - 1; } return l ; } };
好了,今天的文章就到这里,如果觉得有所收获,请顺手点个在看或者转发吧,你们的支持是我最大的动力 。