UPC10544: 凉宫春日的叹息

10544: 凉宫春日的叹息

时间限制: 1 Sec  内存限制: 128 MB
提交: 35  解决: 16
[提交] [状态] [讨论版] [命题人:admin]

题目描述

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

 

输入

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

 

 

输出

仅一个数，表示答案。

 

样例输入

复制样例数据

5 6
1 1 1 1 1

样例输出

2

 

提示

对于15%的数据，n≤1000
对于30%的数据，n≤5000
对于50%的数据，n,k≤105
对于70%的数据，n≤105
对于100%的数据，n≤106,1≤a[i],k≤109

 

 

 

来源/分类

衡水2018NOIP考前集训 

 

#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;
}