题目描述
逆向思维+滑动窗口
题目分析 : 从数组左侧和右侧,取出左侧的连续数字,右侧的连续数字,使得这些数字之和等于 x
,维护最小取数次数,作为答案 。
设整个数组之和 total
,除去左侧和右侧的连续数字,剩下的数组之和等于total-x
,而且剩下的数组正好是中间的连续数字。问题转化为,求数组之和等于total-x
的最长连续数组。中间剩下的数组越长,左侧和右侧使用的数组就越短。
c++
class Solution {
public:
int minOperations(vector<int>& nums, int x) {
int target = accumulate(nums.begin(),nums.end(),0) - x;
if(target<0) return -1;
int ans = -1, l = 0, r = 0 , sum = 0;
while(r<nums.size()){
sum += nums[r++];
while(sum>target) sum -= nums[l++];
if(sum==target) ans = max(r-l,ans);
}
return ans<0?-1:nums.size()-ans;
}
};
python3
class Solution:
def minOperations(self, nums: List[int], x: int) -> int:
target = sum(nums) - x
if target < 0:
return -1
ans = -1
l = r = s = 0
while r < len(nums):
s += nums[r]
r += 1
while s > target:
s -= nums[l]
l += 1
if s == target:
ans = max(ans, r - l)
return ans if ans < 0 else len(nums) - ans
- 时间复杂度 :
O
(
n
)
O(n)
O(n) ,
n
n
n 是数组长度 ,计算
target
,维护滑动窗口的时间复杂度 O ( n ) O(n) O(n) 。 - 空间复杂度 : O ( 1 ) O(1) O(1) , 只使用常量级空间 。
AC
致语
- 理解思路很重要
- 读者有问题请留言,清墨看到就会回复的。