LeetCode 1300. 转变数组后最接近目标值的数组和(二分)

题目

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

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

链接:https://leetcode-cn.com/problems/sum-of-mutated-array-closest-to-target

样例

LeetCode 1300. 转变数组后最接近目标值的数组和(二分)_第1张图片


题解

二分答案

class Solution 
{
     
public:
	int check(vector<int>& arr,int x,int target)
	{
     
		int sum=0;
		for(int i=0;i<arr.size();i++)
		{
     
			if(arr[i]<x) sum+=arr[i];
			else sum+=x; 
		}
		return sum;
	}
    int findBestValue(vector<int>& arr, int target) 
    {
     
    	int sum=0;int high=0;
    	for(int i=0;i<arr.size();i++)
    	{
     
    		sum+=arr[i];high=max(high,arr[i]);
		}
		if(sum<=target) return high;
        int l=0,r=target,mid;
		while(l<r)
		{
     
			if(l+1==r)
			{
     
				int a=check(arr,l,target),b=check(arr,r,target);
				if(abs(target-a)<=abs(target-b)) return l;
				else return r;
			}
			mid=(l+r)/2;
			if(check(arr,mid,target)==target) return mid;
			else if(check(arr,mid,target)<target) l=mid;
			else r=mid;
		} 
    }
};

时间复杂度为 O ( n l o g n ) O(nlogn) O(nlogn)

你可能感兴趣的:(LeetCode 1300. 转变数组后最接近目标值的数组和(二分))