算法
(二分,前缀和) $O(n^2)$
观察每个区间的值 $Y_i = (\sum _j 1) * (\sum _j v_j), j \in [L_i, R_i] 且 w_j \ge W$。
当 $W$ 增大时,区间 $[L_i, R_i]$ 中满足要求的 $(w_i, v_i)$ 会减少,同时所有 $v_i \ge 0$,因此 $Y_i$ 的值也会减少。
由于 $Y = \sum _i Y_i$,所以 $Y$ 随 $W$ 单调递减。
因此我们可以二分出距离 $S$ 最近的值。
剩下的问题是当 $W$ 确定之后,我们如何快速求出每个 $Y_i$。
这里可以预处理出所有满足 $w_i \ge W$ 的元素的前缀和,之后可以 $O(1)$ 时间求出每个区间中所有 $v_i$ 的和,以及满足要求的元素个数。
时间复杂度分析
总共二分 $O(logW)$ 次,每次预处理前缀和的计算量是 $O(N)$,求 $Y$ 的计算量是 $O(M)$。因此总时间复杂度是 $O(N + M)logW)$。
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;
}
就我看不懂这个$Y~i$这个式子吗
你不是一个人
我也是没看懂这个式子,我思路刚开始想到二分了,就是没看懂这个式子所以来yxc大佬的题解爬了
You are not alone
I’m be here with you
You’re not the only one.
我也看了半天没懂……
满足条件得区间长度*每一个满足条件的矿石质量的和
我也差点没看懂你的评论。
应该是每个区间内满足重量w[i] >= W的个数 * 它们价值的总和,就是这个区间的检验值
如果写成 gety(mid) < s然后变成第二种二分模版为什么会超时呢
老师,二分的左边界可以改成1吧,性质是wi>=S的元素个数*元素之和,(wi>0),既然大于等于应该左边界应该可以取1,想确认一下
不可以,可能S很小,最接近的方案是一个矿石都不选,此时
W
要取0。是不是说反了啊,S很小的时候,为了使得S-Y绝对最小,此时W取右边界1e6+1,没有满足要求的,一个都没选,Y是0
S很大的时候,为了使得S-Y绝对值最小,取右边界0,都满足要求,使得Y也很大,都选上