树上游戏（ABC218 G）题解

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({\log }^{2}n)$。