其他分享
首页 > 其他分享> > 凉宫春日的叹息

凉宫春日的叹息

作者:互联网

给定一个数组,将其所有子区间的和从小到大排序,求第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;
}

  

标签:return,叹息,春日,res,ll,mid,int,凉宫,数组
来源: https://www.cnblogs.com/Astronaut0142/p/15004696.html