ABC218G Game on Tree 2（对顶堆）

最新推荐文章于 2023-07-11 21:37:19 发布

tanjunming2020

最新推荐文章于 2023-07-11 21:37:19 发布

阅读量148

点赞数 3

分类专栏：题解文章标签：题解

本文链接：https://blog.csdn.net/tanjunming2020/article/details/131668527

版权

题解专栏收录该内容

348 篇文章 0 订阅

订阅专栏

文章介绍了如何解决一个在树结构上进行的点权中位数游戏。游戏双方A和B轮流移动棋子，目标是最大化或最小化经过的点权中位数。使用对顶堆来动态计算中位数，并在DFS遍历过程中更新每个节点的最佳得分。最终的答案是A从根节点出发能获得的最大得分。

摘要由CSDN通过智能技术生成

ABC218G Game on Tree 2

洛谷ABC218G Game on Tree 2

题目大意

给定一棵树，以及树各节点的点权（点权为偶数）。起初有一个棋子在树的根结点（结点 $1$ ）处。

$A$ 与 $B$ 两人轮流操作， $A$ 先操作，每次操作将棋子移动到其所在节点的某个儿子节点。到某个节点的得分定义为棋子经过所有结点点权的中位数。

当棋子到达某个叶节点时，游戏结束。

$A$ 希望最后得分尽可能大， $B$ 希望最后得分尽可能小。 $A, B$ 都采用最优策略，求最终得分。

题解

前置知识：对顶堆

我们可以先求出每个叶子节点到根节点的各个点的权值的中位数，这就要用到对顶堆了。将这棵树 $df s$ 一遍，走向儿子节点就是往对顶堆中添加元素，走回父亲节点就是往对顶堆中删除元素。删除元素操作不好直接实现，可以对要删除的元素用 $ma p$ 来打个标记，遇到时再删除。

设 $f_u$ 表示走到节点 $u$ 时继续往下走，最终得到的中位数的值。那么，如果当前是 $A$ 移动，则 $f_u$ 取 $u$ 的所有儿子的最大的 $f$ 值；如果当前是 $B$ 移动，则 $f_u$ 取 $u$ 的所有儿子的最小的 $f$ 值。

最后， $f_1$ 就是答案。

时间复杂度为 $O(n\log n)$ 。

code

#include<bits/stdc++.h>
using namespace std;
int n,x,y,tot=0,now,t1,t2,a[100005],f[100005],d[200005],l[200005],r[200005];
map<int,int>z1,z2;
priority_queue<int>v1,v2;
void add(int xx,int yy){
	l[++tot]=r[xx];d[tot]=yy;r[xx]=tot;
}
void gttop1(){
	while(!v1.empty()&&z1[v1.top()]){
		--z1[v1.top()];v1.pop();
	}
}
void gttop2(){
	while(!v2.empty()&&z2[v2.top()]){
		--z2[v2.top()];v2.pop();
	}
}
void maintain(){
	gttop1();gttop2();
	if(t1>=t2+2){
		v2.push(-now);
		now=v1.top();
		v1.pop();
		++t2;--t1;
	}
	else if(t2>=t1+2){
		v1.push(now);
		now=-v2.top();
		v2.pop();
		++t1;--t2;
	}
}
void pt(int k){
	if(k<now){
		++t1;v1.push(k);
	}
	else{
		++t2;v2.push(-k);
	}
	maintain();
}
int gt(){
	gttop1();gttop2();
	if(t1==t2) return now;
	else if(t1>t2) return (now+v1.top())/2;
	else return (now-v2.top())/2;
}
void del(int k){
	if(k<now){
		--t1;++z1[k];
	}
	else if(k>now){
		--t2;++z2[-k];
	}
	else{
		gttop1();gttop2();
		if(t1>t2){
			now=v1.top();
			--t1;v1.pop();
		}
		else{
			now=-v2.top();
			--t2;v2.pop();
		}
	}
	maintain();
}
void dfs(int u,int fa,int dep){
	if(u==1) now=a[u];
	else pt(a[u]);
	if(dep&1) f[u]=0;
	else f[u]=1e9+1;
	for(int i=r[u];i;i=l[i]){
		if(d[i]==fa) continue;
		dfs(d[i],u,dep+1);
		if(dep&1) f[u]=max(f[u],f[d[i]]);
		else f[u]=min(f[u],f[d[i]]);
	}
	if(f[u]==0||f[u]==1e9+1) f[u]=gt();
	if(u>1) del(a[u]);
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	for(int i=1;i<n;i++){
		scanf("%d%d",&x,&y);
		add(x,y);add(y,x);
	}
	dfs(1,0,1);
	printf("%d",f[1]);
	return 0;
}