【LeetCode每日一题:1802. 有界数组中指定下标处的最大值~~~二分+公式】

最新推荐文章于 2023-01-04 23:23:19 发布

硕风和炜

最新推荐文章于 2023-01-04 23:23:19 发布

阅读量99

点赞数 1

分类专栏： LeetCode每日一题打卡文章标签： leetcode 算法 Java 二分查找公式推导

本文链接：https://blog.csdn.net/Coder_ljw/article/details/128544529

版权

LeetCode每日一题打卡专栏收录该内容

421 篇文章 7 订阅

订阅专栏

题目描述

给你三个正整数 n、index 和 maxSum 。你需要构造一个同时满足下述所有条件的数组 nums（下标从 0 开始计数）：

nums.length == n
nums[i] 是正整数，其中 0 <= i < n
abs(nums[i] - nums[i+1]) <= 1 ，其中 0 <= i < n-1
nums 中所有元素之和不超过 maxSum
nums[index] 的值被最大化
返回你所构造的数组中的 nums[index] 。

注意：abs(x) 等于 x 的前提是 x >= 0 ；否则，abs(x) 等于 -x 。

示例 1：

输入：n = 4, index = 2, maxSum = 6
输出：2
解释：数组 [1,1,2,1] 和 [1,2,2,1] 满足所有条件。不存在其他在指定下标处具有更大值的有效数组。
示例 2：

输入：n = 6, index = 1, maxSum = 10
输出：3

提示：

1 <= n <= maxSum <= 109
0 <= index < n

求解思路

这道题目通过二分查找+数学公式推导来求解。
可以使nums[index] 成为数组最大的元素，并使其他元素尽可能小，即从nums[index] 开始，往左和往右，下标每相差 1，元素值就减少 1，直到到达数组边界，或者减少到仅为 1 后保持为 1 不变。
根据这个思路，一旦nums[index] 确定后，这个数组的和numsSum 也就确定了。并且 nums[index] 越大，数组和numsSum 也越大。据此，可以使用二分搜索来找出最大的使得 numsSum≤maxSum 成立的nums[index]。

实现代码

class Solution {
    public int maxValue(int n, int index, int maxSum) {
        int left = 1, right = maxSum;
        while (left <= right) {
            int mid = (right+left) / 2;
            if (valid(mid, n, index, maxSum)) {
                left = mid+1;
            } else {
                right = mid - 1;
            }
        }
        return left-1;
    }

    public boolean valid(int mid, int n, int index, int maxSum) {
        int left = index;
        int right = n - index - 1;
        return mid + cal(mid, left) + cal(mid, right) <= maxSum;
    }

    public long cal(int big, int length) {
        if (length + 1 < big) {
            int small = big - length;
            return (long) (big - 1 + small) * length / 2;
        } else {
            int ones = length - (big - 1);
            return (long) big * (big - 1) / 2 + ones;
        }
    }
}