LeetCode 1477. 找两个和为目标值且不重叠的子数组--双指针双向处理记录前缀

1477. 找两个和为目标值且不重叠的子数组

给你一个整数数组 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

题解：题目难点在于要找到两个互不重叠的子区间，子区间和为target，并且有多个答案时要求输出两个子区间长度和最小的答案。能否把问题简化呢，为了保证绝对不会找到重复的子区间，我从中间一分为2，假设分为两个部分[0,x],[x+1,n-1]两个部分，那么问题就变成了在区间[0,x]和区间[x+1,n-1]之间各找一个子区间，子区间和为target并且长度尽可能小，这样是不是就容易多了，但是麻烦来了，x设置在哪儿呢？还是说暴力将x从0一直到n-1都判断一遍，这样就会超时了，但是发现，如果我用两个数组ml[maxn]和mr[maxn]分别为ml[x]表示区间[0,x]之间得到的满足答案的最小子区间长度，mr[x]表示区间[x+1,n-1]之间得到的满足答案的最小的子区间长度，于是答案变成了ans=min(ans,ml[x]+mr[x+1])，x再从0到n-1跑一遍就行了。
然后ml和mr怎么求呢，就用我们传统的双指针从左到右和从右到左分别跑一遍就找到了，真的是非常amazing啊

AC代码

class Solution {
public:
int ml[100100],mr[100100];
int minSumOfLengths(vector<int>& arr, int target) 
{
	int n=arr.size();
	int l=0,r=0,ans=arr[0];
	for(int i=0;i<n;i++)
	ml[i]=mr[i]=1e9;
	while(r<n)
	{
		if(ans<target)
		{
			r++;
			if(r>=n)break;
			ans+=arr[r];
		}
		else 
		{
			if(ans==target)
			{
				if(r==0)
				{
					ml[r]=r-l+1;
				}
				else
				{
					ml[r]=min(r-l+1,ml[r-1]);
				}		
			}
			ans-=arr[l];
			l++;
		}
		if(r>0)
		ml[r]=min(ml[r],ml[r-1]);
	} 
	ans=arr[n-1];
	l=r=n-1;
	while(l>=0)
	{
		if(ans<target)
		{
			l--;
			if(l<0)break;
			ans+=arr[l];
		}
		else
		{
			if(ans==target)
			{
				if(l==n-1)
				mr[l]=r-l+1;
				else
				{
					mr[l]=min(r-l+1,mr[l+1]);
				}
			}
			ans-=arr[r];
			r--;
		}
		if(l<n-1)
		mr[l]=min(mr[l],mr[l+1]);
	}
	ans=1e9;
	for(int i=0;i<n-1;i++)
	{
		ans=min(ans,ml[i]+mr[i+1]);
	}
	if(ans==1e9)return -1;
	return ans;
}
};