一整数数列a1, a2, ... , an(有正有负),以及另一个整数k,求一个区间[i, j],(1 <= i <= j <= n),使得a[i] + ... + a[j] = k。
Input
第1行:2个数N,K。N为数列的长度。K为需要求的和。(2 <= N <= 10000,-10^9 <= K <= 10^9) 第2 - N + 1行:A[i](-10^9 <= A[i] <= 10^9)。
Output
如果没有这样的序列输出No Solution。 输出2个数i, j,分别是区间的起始和结束位置。如果存在多个,输出i最小的。如果i相等,输出j最小的。
Input示例
6 10 1 2 3 4 5 6
Output示例
1 4
其实不优化可以过,直接计算前缀和来两重for循环暴力枚举即可。
不过如果数据加强了就不行了。
map优化,map储存每一个前缀和的出现次数。
对于某个前缀和,如果当前前缀和加上K 这个值如果也在map当中,那么我们暴力找即可。
如果没有,我们就不找了。直接重新遍历
#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include<math.h>
#include<map>
#include <algorithm>
using namespace std;
long long sum[11000],a[11000];
map<long long ,long long>mp;
int main()
{
int n,k;
cin>>n>>k;
int flag=0;
memset(sum,0,sizeof(sum));
for(int i=1;i<=n;i++)
{
cin>>a[i];
sum[i]=sum[i-1]+a[i];
mp[sum[i]]++;
}
for(int i=0;i<=n;i++)
{
if(mp[sum[i]+k])
{
for(int j=i;j<=n;j++)
{
if(sum[j]-sum[i]==k)
{
cout<<i+1<<" "<<j<<endl;
flag=1;
break;
}
}
}
if(flag)
break;
}
if(!flag)
cout<<"No Solution"<<endl;
}