LeetCode 1658. 将 x 减到 0 的最小操作数

1658. 将 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 。

提示:

  • 1 <= nums.length <= 105
  • 1 <= nums[i] <= 104
  • 1 <= x <= 109

二、方法一

哈希+ 前缀和

class Solution {
    public int minOperations(int[] nums, int x) {
        int len = nums.length;
        Map<Integer, Integer> map = new HashMap<>();
        int[] sum = new int[len + 1];
        map.put(0, 0);
        int temp = 0;
        for (int i = 1; i <= len; i++) {
            map.put(temp = sum[i - 1] + nums[i - 1], i);
            sum[i] = temp;
        }
        int target = sum[len] - x;
        if (target < 0) {
            return -1;
        }
        int res = Integer.MAX_VALUE;
        for (int i = len; i > 0 && sum[i] >= target; i--) {
            if (map.containsKey(temp = sum[i] - target)) {
                res = Math.min(res, len - i + map.get(temp));
            }
        }
        return res == Integer.MAX_VALUE ? -1 : res;
    }
}

复杂度分析

  • 时间复杂度:O(n)。

  • 空间复杂度:O(n)。

你可能感兴趣的:(LeetCode题解,leetcode,算法)