算法的重要性,我就不多说了吧,想去大厂,就必须要经过基础知识和业务逻辑面试+算法面试。所以,为了提高大家的算法能力,这个公众号后续每天带大家做一道算法题,题目就从LeetCode上面选 !
今天和大家聊的问题叫做 排列硬币,我们先来看题面:
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 ;
}
};
好了,今天的文章就到这里,如果觉得有所收获,请顺手点个在看或者转发吧,你们的支持是我最大的动力 。
上期推文: