51nod 1094 和为k的连续区间 map优化

一整数数列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;


}