题目来自LeetCode编号1802
题目如下:
给你三个正整数 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
思路:
对于index位置,要想最大其附近的元素也要与其相差为1,我们可以先给没给位置赋值1,再看看剩余的值够不够给index赋值,如果不够那就直接返回答案
代码如下:
class Solution {
public int maxValue(int n, int index, int maxSum) {
int l=index,r=index;//创建个以index为中心的范围[l,r]
int ans=1;//先给每个位置赋值为1,此时我们的index位置上为
//1,ans就是我们index上的数。
int res =maxSum-n;//查看剩余的数。
while(l>0||r<n-1)//在[l,r]范围赋值
{
int len=r-l+1;//查看[l,r]范围内的数的个数
if(res>=len)//如果剩余的数够在范围内铺一层就进来,否则就不铺了
{
res-=len;//剩余数减去这层数量,得到之后的剩余数
ans++;//铺完一层我们就可以给答案加1
l=Math.max(0,l-1);//l往左边移动扩大范围
r=Math.min(r+1,n-1);//r往右边移动扩大范围
}
else
{
break;
}
}
ans+=res/n;//当处于l,r都到边界时剩余的数,我们就给所有的铺一层
//剩余数不够铺一层就不用往上加了
return ans;
}
}