算法知识点:二分,前缀和

复杂度:

解题思路:

观察每个区间的值

增大时,区间 中满足要求的会减少,同时所有 ,因此 的值也会减少。

由于 ,所以 单调递减。

因此我们可以二分出距离 最近的值。

剩下的问题是当 确定之后,我们如何快速求出每个
这里可以预处理出所有满足 的元素的前缀和,之后可以 时间求出每个区间中所有 的和,以及满足要求的元素个数。

时间复杂度分析:

总共二分 次,每次预处理前缀和的计算量是 ,求 的计算量是 。因此总时间复杂度是

C++ 代码:

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
 
typedef long long LL; const int N = 200010;
 
int n, m;
LL S;
int w[N], v[N];
int l[N], r[N];
int cnt[N];
LL sum[N];
 
LL get(int W)
{
    for (int i = 1; i <= n; i++)
        if (w[i] >= W)
        {
            sum[i] = sum[i - 1] + v[i];
            cnt[i] = cnt[i - 1] + 1;
        }
    else
    {
        sum[i] = sum[i - 1];
        cnt[i] = cnt[i - 1];
    }
 
    LL res = 0;
    for (int i = 0; i < m; i++) res += (cnt[r[i]] - cnt[l[i] - 1]) *(sum[r[i]] - sum[l[i] - 1]);
    return res;
}
 
int main()
{
    scanf("%d%d%lld", &n, &m, &S);
    for (int i = 1; i <= n; i++) scanf("%d%d", &w[i], &v[i]);
    for (int i = 0; i < m; i++) scanf("%d%d", &l[i], &r[i]);
 
    int l = 0, r = 1e6 + 1;
    while (l < r)
    {
        int mid = l + r + 1 >> 1;
        if (get(mid) >= S) l = mid;
        else r = mid - 1;
    }
 
    printf("%lld\n", min(abs(get(r) - S), abs(S - get(r + 1))));
 
    return 0;
}



另外,牛客暑期NOIP真题班限时免费报名
报名链接:https://www.nowcoder.com/courses/cover/live/248
使用优惠券免费报名:DCYxdCJ