LeetCode 5711. 有界数组中指定下标处的最大值（二分+等差数列check）

题意：

给你三个正整数 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 <= maxSum <= 1e9
0 <= index < n

解法：

二分nums[idx]的值mid,考虑如何check:
显然最优情况下,左右两边肯定是mid-1,mid-2...这样递减,是等差数列,
利用等差数列计算出num[idx]取mid时,整个序列的最小总值是sum多少,
如果sum<=maxSum,那么合法,否则不合法.

code：

#define ll long long
class Solution {
public:
    ll cal(int x){
        return 1ll*x*(x+1)/2;
    }
    int check(int mid,int n,int pos,int ma){
        int lc=pos;
        int rc=n-pos-1;
        ll sum=0;
        if(lc<=mid-1)sum+=cal(mid-1)-cal(mid-lc-1);
        else sum+=cal(mid-1)+lc-(mid-1);
        if(rc<=mid-1)sum+=cal(mid-1)-cal(mid-rc-1);
        else sum+=cal(mid-1)+rc-(mid-1);
        sum+=mid;
        return sum<=ma;
    }
    int maxValue(int n, int pos, int ma) {
        int l=1,r=ma;
        int ans=0;
        while(l<=r){
            int mid=(l+r)/2;
            if(check(mid,n,pos,ma))ans=mid,l=mid+1;
            else r=mid-1;
        }
        return ans;
    }
};

---