T1 冰火战士
Description
假如设定比赛温度为 k k k。那么,「有用」的冰系战士,就是 x i ≤ k x_i \le k xi≤k 的这些冰系战士;「有用」的火系战士,就是 x i ≥ k x_i \ge k xi≥k 的这些火系战士。一场比赛的价值,就是冰火两方,「有用」的战士的 y i y_i yi 之和的较小值。具体来说,就是:
min { ∑ i ∈ i c e , x i ≤ k y i , ∑ i ∈ f i r e , x i ≥ k y i } \min\Bigg\{ \sum_{i \in {\rm ice} , x_i \le k}y_i , \sum_{i \in {\rm fire} , x_i \ge k}y_i \Bigg\} min{ i∈ice,xi≤k∑yi,i∈fire,xi≥k∑yi}
我们要选择一个 k k k,使得这个价值尽可能大。如果有多个 k k k 能使价值最大,则选最大的 k k k。
Solution
结论:
答案一定是某个战士的温度,所以可以先把所有战士的温度先离散化,然后在这些值里面找答案。
考虑两个函数,一个是 f i c e ( x ) = ∑ i ∈ i c e , x i ≤ x y i f_{\rm ice}(x)=\sum\limits_{i \in {\rm ice} , x_i \le x}y_i fice(x)=i∈ice,xi≤x∑yi,另一个是 f f i r e ( x ) = ∑ i ∈ f i r e , x i ≥ x y i f_{\rm fire}(x)=\sum\limits_{i \in {\rm fire} , x_i \ge x}y_i ffire(x)=i∈fire,xi≥x∑yi。那么我们就需要取一个 k k k,使得 min { f i c e ( k ) , f f i r e ( k ) } \min\big\{f_{\rm ice}(k),f_{\rm fire}(k)\big\} min{ fice(k),ffire(k)}。
如下图,发现 f i c e ( k ) f_{\rm ice}(k) fice(k) 是随温度升高单调不下降的(本质上是一个前缀和), f f i r e ( k ) f_{\rm fire}(k) ffire(k) 是随温度升高单调不上升的(本质上是一个后缀和),它们的函数图像是一个「 X \rm X X」的模样。
如下图,再设 f ′ ( x ) = min { f i c e ( x ) , f f i r e ( x ) } f'(x)=\min\{f_{\rm ice}(x),f_{\rm fire}(x)\} f′(x)=min{
fice(x),ffire(x)},则它的函数图像应如图中黄线所示,是一个单峰曲线。
发现当 f i c e ( x ) f_{\rm ice}(x) fice(x) 和 f f i r e ( x ) f_{\rm fire}(x) ffire(x)相交时, f ′ ( x ) f'(x) f′(x) 取到最大值。
但是实际上,温度是离散的,因此曲线本质上也是离散的( k k k 只能取整数),
意味着我们可能找不到冰系战士能量和以及火系战士能量和曲线的交点,因此有两个可能的最优 k k k:
- 最大的,使得 f i c e ( k ) < f f i r e ( k ) f_{\rm ice}(k) < f_{\rm fire}(k) fice(k)<ffire(k) 的 k k k
- 最小的,使得 f i c e ( k ) ≥ f f i r e ( k ) f_{\rm ice}(k) \ge f_{\rm fire}(k) fice(k)≥ffire(k) 的 k k k
因为函数上有相同数组成的平台,三分显然行不通,选择二分答案。
考虑用一个支持区间加、单点查询的数据结构(线段树或树状数组,但是树状数组常数较小,于是选择树状数组)维护 f i c e ( x ) f_{\rm ice}(x) fice(x) 和 f f i r e ( x ) f_{\rm fire}(x) ffire(x)。
例如,每个冰系战士,是对所有 k ≥ x i k \ge x_i k≥xi(一段后缀),令它们的 f i c e ( x ) f_{\rm ice}(x) fice(x) 值加上 y i y_i yi ;一个火系战士,是对所有 k ≤ x i k \le x_i k≤xi(一段前缀),令它们的 f f i r e ( x ) f_{\rm fire}(x) ffire(x) 值加上 y i y_i yi。
这样做,时间复杂度为 O ( n log 2 n ) {\mathcal O}(n \log^2 n) O(nlog2n),无法通过本题的数据( 1 ≤ n ≤ 2 × 1 0 6 1\le n\le 2 \times 10^6 1≤n≤2×106)。
考虑优化掉一个 log \log log 。
其实,树状数组上也是可以二分的。
这个「二分」的实现,其实更像「倍增」。
例如用倍增法求 L C A \rm LCA LCA 时,我们从大到小枚举当前节点的 2 log 2 n , ⋯ , 2 0 2^{\log_2 n},\cdots,2^0 2log2n,⋯,20 级祖先,能往上跳就跳。
在树状数组上也是一样,每次枚举从当前点往前跳 2 i 2^i 2i 个位置是否可行。如果可行就跳过去,否则位置不变。
这里的「检查」,其实就是看树状数组上 c u r p o s + 2 i {\rm curpos}+2^i curpos+2i 的这个位置里填的数(树状数组中 c i c_i ci 填的是 [ i − l o w b i t ( i ) + 1 , i ] [i-{\rm lowbit}(i)+1,i] [i−lowbit(i)+1,i]这段区间的信息, c u r p o s + 2 i {\rm curpos}+2^i curpos+2i 这个位置填的就是 [ c u r p o s + 1 , c u r p o s + 2 i ] [{\rm curpos}+1,{\rm curpos}+2^i] [curpos+1,curpos+2i] 这段的信息)。
查询的时候,先二分出「最大的,使得 f i c e ( k ) < f f i r e ( k ) f_{\rm ice}(k) < f_{\rm fire}(k) fice(k)<ffire(k) 的 k k k」和「最小的,使得 f i c e ( k ) ≥ f f i r e ( k ) f_{\rm ice}(k) \ge f_{\rm fire}(k) fice(k)≥ffire(k) 的 k k k」,并查询出它们的 f f f 值。如果第二个 f f f 值更大,则带入这个值,再二分一次,时间复杂度为 O ( n log n ) {\mathcal O}(n \log n) O(nlogn)。
Code
#define int long long
int n,cnt,sum,a[2000001];
struct Query {
int p,t,x,y;
}q[2000001];
//树状数组基本操作
struct BIT {
int n,T[2000001];
#define lowbit(x) (x)&(-x)
void add(int x,int y) {
for (int i=x;i<=n;i+=lowbit(i)) T[i]+=y;}
//修改
int query(int x) {
int sum=0;
for (int i=x;i;i-=lowbit(i)) sum+=T[i];
return sum;
}
//查询
}T1,T2;//ice的BIT和fire的BIT
signed main() {
scanf("%lld",&n);
for (int i=1;i<=n;i++) {
scanf("%lld%lld",&q[i].p,&q[i].t);
if (q[i].p==1) {
scanf("%lld%lld",&q[i].x,&q[i].y);
a[++cnt]=q[i].x;
}
}
sort(a+1,a+cnt+1);
cnt=unique(a+1,a+