给你一个整数数组 arr 和一个整数值 target 。
请你在 arr 中找 两个互不重叠的子数组 且它们的和都等于 target 。可能会有多种方案,请你返回满足要求的两个子数组长度和的 最小值 。
请返回满足要求的最小长度和,如果无法找到这样的两个子数组,请返回 -1 。
示例 1:
输入:arr = [3,2,2,4,3], target = 3
输出:2
解释:只有两个子数组和为 3 ([3] 和 [3])。它们的长度和为 2 。
示例 2:
输入:arr = [7,3,4,7], target = 7
输出:2
解释:尽管我们有 3 个互不重叠的子数组和为 7 ([7], [3,4] 和 [7]),但我们会选择第一个和第三个子数组,因为它们的长度和 2 是最小值。
示例 3:
输入:arr = [4,3,2,6,2,3,4], target = 6
输出:-1
解释:我们只有一个和为 6 的子数组。
示例 4:
输入:arr = [5,5,4,4,5], target = 3
输出:-1
解释:我们无法找到和为 3 的子数组。
示例 5:
输入:arr = [3,1,1,1,5,1,2,1], target = 3
输出:3
解释:注意子数组 [1,2] 和 [2,1] 不能成为一个方案因为它们重叠了。
提示:
1 <= arr.length <= 10^5
1 <= arr[i] <= 1000
1 <= target <= 10^8
我们设dp[i]表示前i个元素中存在的连续子数组和为target的最小长度。
由于数组中所有的数都为正数,所以我们可以利用hash值来快速查询以索引i为结尾且和为target的连续子数组是否存在,如果存在则它的长度肯定是唯一的,我们可以利用hash[sum-target]]快速的得该连续子数组的最左的起点位置索引j,即可快速求出连续子数组的长度.我们用len(target,i)代表和为target且最右边的终点为索引i的连续子数组的长度。则可以得到公式为:
minSum=min(minSum,len(target,i)+dp[i−len(target,i)])(iflen(target,i)>0)
dp[i]=min(dp[i−1],len(target,i))(iflen(target,i)>0)
代码:
class Solution {
public:
int minSumOfLengths(vector<int>& arr, int target) {
int n = arr.size();
int res = n+1;
unordered_map<int,int> cnt;
vector<int> dp(n,n+1);
cnt[0] = -1;
int sum = 0;
for(int i = 0; i < arr.size(); ++i){
sum += arr[i];
cnt[sum] = i;
if(i > 0) dp[i] = dp[i-1];
if(cnt.count(sum-target)){
int prev = cnt[sum-target];
if(prev >= 0 && dp[prev] <= n){
res = min(i - prev + dp[prev],res);
}
dp[i] = min(dp[i],i - prev);
}
}
if(res > n) return -1;
return res;
}
};