ABC218(D~G)题解
主要思路其他题解都有,不说了,主要说一下维护中位数的方法。
这里也可以用树状数组+二分维护。树状数组的基本操作大家应该都会,就不赘述了。
首先把所有点的权值离散化,每次插入一个值就给对应的点加一,这样能够更快的求出比某一个值小的元素数量。
接下来就是二分查找中位数,这也是比较简单的,代码如下。
LL l=1,r=m;//m是离散化之后的元素个数
while(l<r){
LL mid=(l+r)>>1;
if(query(mid)>=(dep[x]+1)/2)r=mid;//(dep[x]+1)/2是当前中位数的排名
else l=mid+1;
}
如果是奇数个数,查找到这里就可以了,但是偶数不行,还要找出另一个数取平均。
同样的二分:
if(query(r)>dep[x]/2)f[x]=c[r];
//如果上一次查找找到的数,比它小的数个数大于中位数(更小的那个)排名,
//那么中位数一定是两个一样的数。
else{
LL temp=r;
l=r+1;r=m;
while(l<r){
LL mid=(l+r)>>1;
if(query(mid)-query(temp))r=mid;//找到离更小的中位数最近的且比它更大的数
else l=mid+1;
}
f[x]=(c[r]+c[temp])/2;
}
树状数组时间复杂度 O ( log n ) O(\log n) O(logn),二分 O ( log n ) O(\log n) O(logn),所以维护时间复杂度 O ( log 2 n ) O(\log^2 n) O(log2n)。