1094 和为k的连续区间

1094 和为k的连续区间

一整数数列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
想看几个小时，发现可以用前缀和+二分来做,只要有sum[j]-sum[i-1] = k，答案答案就是i j 了。

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int N = 10010;
ll a[N], sum[N];
ll n, k;
struct Nod{
    ll sum;
    int id;
    Nod(){
        sum = 0;
    }
}nod[N];
bool cmp(Nod a, Nod b) {
    if(a.sum != b.sum) return a.sum < b.sum;
    else a.id < b.id;
}
int bin(ll x) {
    int l = 1, r = n, pos = -1;
    while(l < r) {
        int m = (l + r) >> 1;
        if(nod[m].sum < x) {
            l = m + 1;
            pos = l;
        }
        else {
            r = m;
            pos = r;
        } 
    }
    return pos;
}
int main() {
    cin >> n >> k;
    for(int i = 1; i <= n; i ++) {
        scanf("%lld", &a[i]);
        sum[i] = sum[i-1] + a[i];
        nod[i].sum = nod[i-1].sum + a[i];
        nod[i].id = i;
    }
    sort(nod+1, nod+1+n, cmp);
    bool flag = false;
    for(int i = 1; i <= n; i ++) {
        ll sumj = sum[i-1] + k;
        int index = bin(sumj);
        // printf("%lld %d %d\n",sumj,i,index );
        if(sum[nod[index].id] != sumj) continue;
        while(nod[index].sum == sumj && nod[index].id < i) index++;
        if(nod[index].id >= i && nod[index].sum == sumj) {
            printf("%d %d\n",i,nod[index].id);
            flag = true;
            break;
        }
    }
    if(!flag) printf("No Solution\n");
    return 0;
}