CD 9 10 11 前缀和问题

题目描述
给定一个无序数组arr, 其中元素可正、可负、可0。给定一个整数k，求arr所有子数组中累加和为k的最长子数组长度
输入描述:
第一行两个整数N, k。N表示数组长度，k的定义已在题目描述中给出
第二行N个整数表示数组内的数
输出描述:
输出一个整数表示答案
示例1
输入
5 0
1 -2 1 1 1
输出
3

 #include<iostream>
 #include<algorithm>
 #include<map>
 using namespace std;
 typedef long long ll;
 map<int,int> mp;//前缀和-》位置 
 int a[100007];//数组可以省略更省空间
 int main()
 {
 	int n,k;
 	scanf("%d%d",&n,&k);
 	ll res=0;
	int len=0;
 	for(int i=0;i<n;i++)
	 {
	 	scanf("%d",&a[i]);
	 	res+=a[i];
	 	if( res==k )
		 	len=max(len,i+1);
	 	if( mp.find(res-k)!=mp.end() )
		 {
		 	len= max( len , i- mp.find(res-k)->second ) ;
		 }
		if( mp.find(res)==mp.end() )
		{
			mp[res]=i;
		}
	 } 
	 cout<<len<<endl;
	  return 0;
 }

题目描述
给定一个无序数组arr，其中元素可正、可负、可0。求arr所有子数组中正数与负数个数相等的最长子数组的长度。
[要求]
时间复杂度为O(n)O(n)，空间复杂度为O(n)O(n)
输入描述:
第一行一个整数N，表示数组长度
接下来一行有N个数表示数组中的数
输出描述:
输出一个整数表示答案
示例1
输入
复制
5
1 -2 1 1 1
输出
复制
2

 #include<iostream>
 #include<algorithm>
 #include<map>
 using namespace std;
 map<int,int> mp;//前缀和-》位置 
 
 int main()
 {
 	int n,k;
 	scanf("%d",&n);
 	int res=0;//表示正数和负数的个数
 	int len=0,x=0;
	mp[0]=-1; 
    for(int i=0;i<n;i++)
	{
		scanf("%d",&x);
		if(x>0)
		   res++;
		else if(x<0)
		   res--;
		if( mp.find(res)!=mp.end() )
			len=max( len , i-mp.find(res)->second ) ;
		else
		   mp[res]=i;
	}
	cout<<len<<endl;
	
 	return 0;
 }

题目描述
给定一个无序数组arr，其中元素只能是1或0。求arr所有的子数组中0和1个数相等的最长子数组的长度
[要求]
时间复杂度为O(n)O(n)，空间复杂度为O(n)O(n)
输入描述:
第一行一个整数N，表示数组长度
接下来一行有N个数表示数组中的数
输出描述:
输出一个整数表示答案
示例1
输入
复制
5
1 0 1 0 1
输出
复制
4

 #include<iostream>
 #include<algorithm>
 #include<map>
 using namespace std;
 
 map<int,int> mp;//前缀和->位置 
 
 int main()
 {
 	int n,k;
 	scanf("%d",&n);
 	int res=0;//表示正数和负数的个数
 	int len=0,x=0;
	mp[0]=-1; 
    for(int i=0;i<n;i++)
	{
		scanf("%d",&x);
		res= x==1? res+1:res-1;
		if( mp.find(res)!=mp.end() )
			len=max( len , i-mp.find(res)->second ) ;
		else
		   mp[res]=i;
	}
	cout<<len<<endl;
	
 	return 0;
 }