凉宫春日的叹息

给定一个数组，将其所有子区间的和从小到大排序，求第k小的是多少。

输入
第一行两个数n, k，表示数组的长度和k；
第二行有n个数，第i个是a[i]，表示给定的数组。

输出
仅一个数，表示答案。

样例输入
5 6
1 1 1 1 1

样例输出
2

二分答案，逐一寻找即可，计算所有小于的，如果不行，那么就在另外一边

#include "bits/stdc++.h"

using namespace std;

typedef long long ll;
const int maxn = 1e6 + 1000;
ll a[maxn];
int n, k;

ll check(ll mid) {
    int j = 0;
    ll res = 0;
    for (int i = 1; i <= n; i++) {
        while (a[i] - a[j] > mid) j++;
        res += i - j;
    }
    return res;
}

int main() {
    //freopen("input.txt", "r", stdin);
    scanf("%d %d", &n, &k);
    for (int i = 1; i <= n; i++) {
        scanf("%lld", &a[i]);
        a[i] += a[i - 1];
    }
    ll l = 0, r = a[n], mid;
    while (l < r) {
        mid = (l + r) >> 1;
        if (check(mid) >= k) r = mid;
        else l = mid + 1;
    }
    printf("%lld\n", r);
    return 0;
}