Java教程

凉宫春日的叹息

本文主要是介绍凉宫春日的叹息,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

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

  

这篇关于凉宫春日的叹息的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!