题目:
给你一个整数数组 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;
}
}