和为k的连续区间（51Nod-1094）

题目

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

思路：求出前缀和后，然后查询 sum[r]-sum[l] 是否为 k 即可，若为 k 则记录区间位置

源程序

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<utility>
#include<stack>
#include<queue>
#include<vector>
#include<set>
#include<map>
#define E 1e-9
#define PI acos(-1.0)
#define INF 0x3f3f3f3f
#define LL long long
const int MOD=7;
const int N=10000+5;
const int dx[]= {-1,1,0,0};
const int dy[]= {0,0,-1,1};
using namespace std;
LL a[N];
LL sum[N];
void init(int n){
    for(int i=1;i<=n;i++)
        sum[i]=sum[i-1]+a[i];
}
LL query(int L,int R){
    return sum[R]-sum[L-1];
}
int main(){
    int n,k;
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
        scanf("%lld",&a[i]);

    init(n);
    bool flag=false;
    int left,right;
    for(int i=1;i<=n;i++){
        for(int j=i;j<=n;j++){
            if(query(i,j)==k){
                left=i;
                right=j;
                flag=true;
                break;
            }
        }
        if(flag)
            break;
    }
    if(flag)
        printf("%d %d\n",left,right);
    else
        printf("No Solution\n");

    return 0;
}