1521. 找到最接近目标值的函数值(双指针+与操作)

题目传送门

视频讲解

 

 


Winston 构造了一个如上所示的函数 func 。他有一个整数数组 arr 和一个整数 target ,他想找到让 |func(arr, l, r) - target| 最小的 l 和 r 。

请你返回 |func(arr, l, r) - target| 的最小值。

请注意, func 的输入参数 l 和 r 需要满足 0 <= l, r < arr.length 。

 

示例 1:

输入:arr = [9,12,3,7,15], target = 5
输出:2
解释:所有可能的 [l,r] 数对包括 [[0,0],[1,1],[2,2],[3,3],[4,4],[0,1],[1,2],[2,3],[3,4],[0,2],[1,3],[2,4],[0,3],[1,4],[0,4]], Winston 得到的相应结果为 [9,12,3,7,15,8,0,3,7,0,0,3,0,0,0] 。最接近 5 的值是 7 和 3,所以最小差值为 2 。
示例 2:

输入:arr = [1000000,1000000,1000000], target = 1
输出:999999
解释:Winston 输入函数的所有可能 [l,r] 数对得到的函数值都为 1000000 ,所以最小差值为 999999 。
示例 3:

输入:arr = [1,2,4,8,16], target = 0
输出:0
 

这个题首先想到的是三层for循环,枚举l,r然后再算这个&这样肯定是要超时的。。。。。然后想一想怎么优化

优化1.首先这个&是不能用前缀和的因为他没有减法,但是你可以用一个前缀和每一位1的个数来优化它,但是这样还是n^2的复杂度

优化2.这个与是有单调性的就是a^x<=a的所以可以用双指针算法的,就是这个l     l1           r     r1,如果r->r1那么l一定是向l1去移动的就是这样

经过这个个优化可以优化成n*20的复杂度

 

for(int i=1,j=1;i<=n;i++){
            while(j+1<=i&&get_sum(j+1,i)<=target) j++;
            ans=min(ans,abs(get_sum(j,i)-target));//此时的i,j是满足条件的<=target 
            if(j<i) ans=min(ans,abs(get_sum(j+1,i)-target));//由于是绝对值,i+1,j肯定是大于target的, 
        }          

 

 

class Solution {
public:
    vector<vector<int>> s;
    int get_sum(int l,int r){
        int ans=0;
        for(int i=0;i<20;i++){
            if(s[i][r]-s[i][l-1]==0){
                ans+=1<<i;
            }
        }
        return ans;    
    }
    int closestToTarget(vector<int>& arr, int target) {
        int n=arr.size();
        s = vector<vector<int>>(20, vector<int>(n + 1));
        for(int i=0;i<20;i++){
            for(int j=1;j<=n;j++){
                s[i][j]=s[i][j-1];
                if(!((arr[j-1]>>i)&1)){
                    s[i][j]++;
                }
            }
        }
        int ans=INT_MAX;
        for(int i=1,j=1;i<=n;i++){
            while(j+1<=i&&get_sum(j+1,i)<=target) j++;
            ans=min(ans,abs(get_sum(j,i)-target));
            if(j<i) ans=min(ans,abs(get_sum(j+1,i)-target));
        }
        return ans;
    }
};

 

posted @ 2021-05-28 11:06  lipu123  阅读(103)  评论(0)    收藏  举报