将 x 减到 0 的最小操作数

将 x 减到 0 的最小操作数

题目:
给你一个整数数组 nums 和一个整数 x 。每一次操作时,你应当移除数组 nums 最左边或最右边的元素,然后从 x 中减去该元素的值。请注意,需要 修改 数组以供接下来的操作使用。

如果可以将 x 恰好 减到 0 ,返回 最小操作数 ;否则,返回 -1 。

示例 1:

输入:nums = [1,1,4,2,3], x = 5
输出:2
解释:最佳解决方案是移除后两个元素,将 x 减到 0 。
示例 2:

输入:nums = [5,6,7,8,9], x = 4
输出:-1
示例 3:

输入:nums = [3,2,20,1,1,3], x = 10
输出:5
解释:最佳解决方案是移除后三个元素和前两个元素(总共 5 次操作),将 x 减到 0 。

解题思路:首先求出数组的前缀和,题目求左右两边的和等于x,也就是当中间元素的和等于(数组前缀和-x)即可

class Solution {
    public int minOperations(int[] nums, int x) {
        int sum = 0;
        int ans = -1;
        for(int num : nums) {
            sum += num;
        }
        
        if(sum < x)
            return -1;
        
        if(sum == x)
            return nums.length;
        
        int r = 0, l = 0, target = sum - x, tmp = 0;
        while(l < nums.length && r < nums.length) {
            while(r < nums.length && tmp <= target) {
                tmp += nums[r++];
            }
            
            //比较中间元素个数,当中间元素个数最多时也就是左右两边元素删去最少
            if(tmp - nums[r - 1] == target) {
                ans = Math.max(ans, r - l - 1);
            }
            
            while(l < r && tmp > target) {
                tmp -= nums[l++];
            }
            
            if(tmp == target) {
                ans = Math.max(ans, r - l);
            }
        }
        
        return ans == -1 ? -1 : nums.length - ans;
    }
}

你可能感兴趣的:(leetcode-数组)