51Nod 1094 和为k的连续区间 前缀和+map应用

一整数数列a1, a2, ... , an（有正有负），以及另一个整数k，求一个区间[i, j]，(1 <= i <= j <= n)，使得a[i] + ... + a[j] = k。

收起

输入

第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)。

输出

如果没有这样的序列输出No Solution。
输出2个数i, j，分别是区间的起始和结束位置。如果存在多个，输出i最小的。如果i相等，输出j最小的。

输入样例

6 10
1
2
3
4
5
6

输出样例

1 4

 先求出1-i的前缀和，然后用map储存起来。

因为 sum[r]-sum[l]=k;

所以之后求是否在map中存在k+sum[i]。若存在则按从小到大查找sum[j]-s[i]==k

代码如下：
 

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <map>
using namespace std;
const int maxn=10005;
typedef long long ll;
ll a[maxn],sum[maxn];
ll k;
int n;
map<ll,ll>ma;
int main()
{
    memset (sum,0,sizeof(sum));
    scanf("%d%lld",&n,&k);
    for (int i=1;i<=n;i++)
    {
        scanf("%lld",&a[i]);
        sum[i]=sum[i-1]+a[i];
        ma[sum[i]]=1;
    }
    int flag=0;
    for (int i=0;i<=n;i++)
    {
        if(ma[sum[i]+k])
        {
            for (int j=i;j<=n;j++)
            {
                if(sum[j]-sum[i]==k)
                {
                    flag=1;
                    printf("%d %d\n",i+1,j);
                    return 0;
                }
            }
        }
    }
    if(!flag)
    {
        printf("No Solution\n");
    }
    return 0;
}