51NOD 1094 和为k的连续区间(前缀和 + map)

传送门

1094 和为k的连续区间
基准时间限制:1 秒 空间限制:131072 KB
一整数数列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

解题思路:
这个题目可以借助前缀和和map来做,首先我们定义

sum[i]=0i1ai
然后我们就可以每次判断了其中用一个map,再求前缀和的时候将map[sum[i]]++,最后就是判断map[sum[i]+k]是不是0就行了。
My Code:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <map>
using namespace std;
typedef long long LL;
const int MAXN = 1e4+5;
LL a[MAXN], sum[MAXN];
map<LL,LL>mp;
int main()
{
    int N, K, i, j;
    while(~scanf("%d%d",&N,&K))
    {
        memset(sum, 0, sizeof(sum));
        for(i=1; i<=N; i++)
        {
            scanf("%lld",&a[i]);
            sum[i] = sum[i-1] + a[i];
            mp[sum[i]]++;
        }
        for(i=0; i<=N; i++)
            if(mp[sum[i]+K])
                for(j=i; j<=N; j++)
                    if(sum[j]-sum[i] == K)
                    {
                        printf("%d %d\n",i+1, j);
                        goto endW;
                    }
        puts("No Solution");
        endW:;
    }
    return 0;
}

你可能感兴趣的:(51NOD 1094 和为k的连续区间(前缀和 + map))