1300. 转变数组后最接近目标值的数组和

给你一个整数数组 arr 和一个目标值 target ,请你返回一个整数 value ,使得将数组中所有大于 value 的值变成 value 后,数组的和最接近 target (最接近表示两者之差的绝对值最小)。

如果有多种使得和最接近 target 的方案,请你返回这些整数中的最小值。

请注意,答案不一定是 arr 中的数字。

示例 1:

输入:arr = [4,9,3], target = 10
输出:3
解释:当选择 value 为 3 时,数组会变成 [3, 3, 3],和为 9 ,这是最接近 target 的方案。
示例 2:

输入:arr = [2,3,5], target = 10
输出:5
示例 3:

输入:arr = [60864,25176,27249,21296,20204], target = 56803
输出:11361

提示:

1 <= arr.length <= 10^4
1 <= arr[i], target <= 10^5

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sum-of-mutated-array-closest-to-target
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路:
设S(i)=sum[i]+(len-1-i)*arr[i],易知S(i)是单调非减函数
而要想最接近目标值,无非就是找最大的pos使得S(pos)<target和最小的pos使得S(pos)>=target
而这两个pos很明显是挨着的
那么二分查找最大的pos(以arr[pos]为value,使得sum[pos]+(len-1-pos)*arr[pos]<target)
如果能找到这样一个pos,那么value一定在[ arr[pos] , arr[pos+1] ]范围内
需要分配的值就是target-sum[pos](设为c),要分配的份数就是len-1-pos(设为num)
设a=target/c,b=a+1,然后检验这两个数是否合法(是否大于等于arr[pos]且小于等于arr[pos+1])
如果合法就比较谁更接近即可

class Solution {
    public int findBestValue(int[] arr, int target) {
        int len=arr.length;
        Arrays.sort(arr);//排序
        int[] sum=new int[len];
        sum[0]=arr[0];
        for(int i=1;i<len;i++){//求前缀和,方便检验
            sum[i]=sum[i-1]+arr[i];
        } 
        int ans=arr[len-1];//初值设为数组内最大的数,再大没有意义
        int mn=Math.abs(sum[len-1]-target);
        int l=0,r=len-2,mid;//二分查找最大的pos(以arr[pos]为value,使得sum[pos]+(len-1-pos)*arr[pos]<target)
        while(l<=r){
            mid=(l+r)>>1;
            if(sum[mid]+(len-1-mid)*arr[mid]<target) l=mid+1;
            else r=mid-1;
        }
        int pos=l-1;//如果pos>=0,那么value的值一定在[arr[pos],arr[pos+1]]范围内
        if(pos<0){
            int a=target/len;//向下取整
            int b=a+1;//如果不整除就是向上取整,否则就无效,会在范围判断中去掉
            if(a<=arr[0]){
                if(Math.abs(a*len-target)<mn){
                    mn=Math.abs(a*len-target);
                    ans=a;
                }
                else if(Math.abs(a*len-target)==mn&&a<ans){
                    ans=a;
                }
            }
            if(b<=arr[0]){
                if(Math.abs(b*len-target)<mn){
                    mn=Math.abs(b*len-target);
                    ans=b;
                }
                else if(Math.abs(b*len-target)==mn&&b<ans){
                    ans=b;
                }
            }
        }
        else{
            int num=len-1-pos;
            int c=target-sum[pos];//需要平均分配的值
            int a=c/num;//向下取整
            int b=a+1;//如果不整除就是向上取整,否则就无效,会在范围判断中去掉
            if(a>=arr[pos]&&a<=arr[pos+1]){
                if(Math.abs(sum[pos]+a*num-target)<mn){
                    mn=Math.abs(sum[pos]+a*num-target);
                    ans=a;
                }
                else if(Math.abs(sum[pos]+a*num-target)==mn&&a<ans){
                    ans=a;
                }
            }
            if(b>=arr[pos]&&b<=arr[pos+1]){
                if(Math.abs(sum[pos]+b*num-target)<mn){
                    mn=Math.abs(sum[pos]+b*num-target);
                    ans=b;
                }
                else if(Math.abs(sum[pos]+b*num-target)==mn&&b<ans){
                    ans=b;
                }
            }
        }
        return ans;
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值