CSP2019 树的重心 题解
本题当然可以通过大力讨论每棵子树的size的大小关系,然后用各种数据结构暴力维护。但是我更倾向于用一种更为性质的做法。
首先讲一下我在考场上想到的做法(没写)。就是考虑换根,在换根的过程中计算每一条边删去后得到的两棵子树的重心, 由于重心的一些性质,如果我们把以点v的所有儿子为根的子树的重心求了出来,那么我们要求以点v为根的子树的重心时,可以保证重心一定在v的重儿子子树的重心到v的链上,那么我们就可以通过链上倍增来实现对多棵子树的重心的合并。在换根时维护每个点子树size,以及子树内的重心,换根时倍增求新子树的重心即可。
代码:
#include<bits/stdc++.h>
using namespace std;
#define N 300007
#define M 600007
#define ll long long
#define mem(x) memset(x,0,sizeof(x))
int hd[N],pre[M],to[M],num,wa[N],wb[N],sz[N],ts,dep[N];
int anc[N][23],ms[N],n;
ll ans;
void clear()
{
mem(hd),mem(wa),mem(wb),mem(sz);
mem(dep),mem(ms);
num=0;ans=0;
}
int calc(int v)
{
return max(ms[v],ts-sz[v]);
}
void solve(int x,int d,int &a,int &b)
{
if(dep[x]<d)return;
for(int i=20;i>=0;i--)
{
int mid=anc[x][i],y=anc[mid][0];
if(dep[y]>=d&&calc(mid)>=calc(y))
x=mid;
}
int y=anc[x][0];
int dx=calc(x),dy=calc(y);
if(dep[x]==d||dx<dy)
{
if(dx<=ts/2)a=x;
}
else
{
if(dy<=ts/2)
{
a=y;
if(dx==dy)b=x;
}
}
}
/*void solve(int x,int d,int &a,int &b)
{
while(dep[x]>=d)
{
if(calc(x)<=ts/2)
{
if(a)b=x;
else a=x;
}
x=anc[x][0];
}
}*/
int glca(int x,int y)
{
if(dep[x]<dep[y])swap(x,y);
for(int i=20;i>=0;i--)
if(dep[x]-dep[y]>=(1<<i))
x=anc[x][i];
if(x==y)return x;
for(int i=20;i>=0;i--)
if(anc[x][i]!=anc[y][i])
x=anc[x][i],y=anc[y][i];
return anc[x][0];
}
int dmax(int x,int y)
{
return dep[x]>dep[y]?x:y;
}
int dmin(int x,int y)
{
return dep[x]<dep[y]?x:y;
}
void dfs(int v,int f)
{
dep[v]=dep[f]+1;
anc[v][0]=f;
for(int i=1;i<=20;i++)
anc[v][i]=anc[anc[v][i-1]][i-1];
sz[v]=1;
int son=v;
for(int i=hd[v];i;i=pre[i])
{
int u=to[i];
if(u==f)continue;
dfs(u,v);
sz[v]+=sz[u];
if(sz[u]>ms[v])ms[v]=sz[u],son=u;
}
ts=sz[v];
int x=( son==v ? v : dmax(wa[son],wb[son]) );
solve(x,dep[v],wa[v],wb[v]);
}
void dp(int v,int f)
{
int s1,d1,s2,d2;
s1=d1=s2=d2=0;
for(int i=hd[v];i;i=pre[i])
{
int u=to[i];
if(sz[u]>s1)s2=s1,d2=d1,s1=sz[u],d1=u;
else if(sz[u]>s2)s2=sz[u],d2=u;
}
for(int i=hd[v];i;i=pre[i])
{
int u=to[i];
if(u==f)continue;
int x=d1;
if(u==d1)x=d2;
int _ms=ms[v],_wa=wa[v],_wb=wb[v];
sz[v]-=sz[u],sz[u]+=sz[v];
ms[v]=sz[x];
wa[v]=wb[v]=0;
ts=sz[v];
int p=dmax(wa[x],wb[x]);
//if(p==0)printf("WA\n");
if(x==f)
{
int q=dmin(wa[x],wb[x]);
int lca=glca(p,v);
if(q!=0&&q==anc[lca][0])p=q,lca=p;
solve(p,dep[lca],wa[v],wb[v]);
solve(v,dep[lca]+1,wb[v],wa[v]);
}
else
{
solve(p,dep[v],wa[v],wb[v]);
}
/*if(!wa[v]&&!wb[v])
{
printf("%d\n",v);
printf("%d\n",x);
}*/
ans+=wa[v]+wb[v]+wa[u]+wb[u];
dp(u,v);
wa[v]=_wa,wb[v]=_wb;
ms[v]=_ms;
sz[u]-=sz[v],sz[v]+=sz[u];
}
}
void adde(int x,int y)
{
num++;pre[num]=hd[x];hd[x]=num;to[num]=y;
}
void dewa()
{
for(int i=1;i<=n;i++)
printf("%d: %d %d\n",i,wa[i],wb[i]);
}
int main()
{
//freopen("data.in","r",stdin);
int t;
scanf("%d",&t);
while(t--)
{
clear();
int x,y;
scanf("%d",&n);
for(int i=1;i<n;i++)
{
scanf("%d%d",&x,&y);
adde(x,y),adde(y,x);
}
dfs(1,0);
//dewa();
dp(1,0);
printf("%lld\n",ans);
}
return 0;
}
还有一种题解做法,也是利用重心的性质,但是比我的更巧妙,而且实现也更简单。
就是考虑以某一个点v为根时怎么找它的重心,可以发现从点v出发,到重心的路径一定是不断沿着重儿子走,而每一个点的重儿子是唯一的,那么我们用倍增优化沿着重儿子走的过程,设f[i][j]为从点i开始,走\(2^j\)步走到的位置,我们只要找到最后一个满足\(2*size[v]\geq sum\)的点即可,其中sum是割边后当前连通块的大小。同样像前一种做法一样换根即可,注意f[i][j]是可以换根维护的。
题解做法:
#include<bits/stdc++.h>
using namespace std;
#define N 300007
#define M 600007
#define ll long long
#define mem(x) memset(x,0,sizeof(x))
int hd[N],pre[M],to[M],num,son[N],fa[N],sz[N];
int ch[N][23],n;
ll ans;
void clear()
{
mem(hd),mem(son);
ans=0,num=0;
}
void adde(int x,int y)
{
num++;pre[num]=hd[x];hd[x]=num;to[num]=y;
}
void mak_st(int v)
{
for(int i=1;i<=20;i++)
ch[v][i]=ch[ch[v][i-1]][i-1];
}
void dfs(int v,int f)
{
fa[v]=f,sz[v]=1;
for(int i=hd[v];i;i=pre[i])
{
int u=to[i];
if(u==f)continue;
dfs(u,v);
sz[v]+=sz[u];
if(sz[u]>sz[son[v]])son[v]=u;
}
ch[v][0]=son[v];
mak_st(v);
}
void solve(int x,int ts)
{
for(int i=20;i>=0;i--)
{
int y=ch[x][i];
if(y&&2*sz[y]>=ts)x=y;
}
ans+=x;
if(2*sz[x]==ts)ans+=fa[x];
}
void dp(int v,int f)
{
int s1,s2;
s1=s2=0;
for(int i=hd[v];i;i=pre[i])
{
int u=to[i];
if(sz[u]>sz[s1])s2=s1,s1=u;
else if(sz[u]>sz[s2])s2=u;
}
for(int i=hd[v];i;i=pre[i])
{
int u=to[i];
if(u==f)continue;
int x=s1;
if(u==s1)x=s2;
sz[v]-=sz[u];
ch[v][0]=x;
mak_st(v);
solve(v,sz[v]),solve(u,sz[u]);
sz[u]+=sz[v],fa[v]=u;
dp(u,v);
sz[u]-=sz[v],sz[v]+=sz[u];
}
ch[v][0]=son[v];
mak_st(v);
fa[v]=f;
}
int main()
{
//freopen("data.in","r",stdin);
int t;
scanf("%d",&t);
while(t--)
{
clear();
scanf("%d",&n);
int x,y;
for(int i=1;i<n;i++)
{
scanf("%d%d",&x,&y);
adde(x,y),adde(y,x);
}
dfs(1,0),dp(1,0);
printf("%lld\n",ans);
}
return 0;
}
比较两种做法,发现其实都利用的是重儿子的性质,只不过我的做法是从重儿子子树内往根走,而题解做法是从根往重儿子走。为什么题解做法比我的短了很多呢?
因为我的倍增数组是从下往上倍增,换根时根的改变会导致下面所有子孙结点的倍增数组改变,于是不能在换根过程中维护,所以我的倍增数组一直是以1号点为根的,就多了当前点的重儿子是从以1为根时的父亲方向来还是从儿子方向来的讨论。而题解是从上往下倍增的,换根时不会导致下面的点发生改变,而能够自上往下倍增的原因又是题解利用了每个点只会往唯一的重儿子走的性质,那么每个点出发走若干步走到的点是唯一的,就可以倍增了。
我在洛谷的题解中,看到了一篇大力讨论子树size的题解,里面的一些计数技巧还是有点意思的,比如说把计算每条边割断后子树重心标号之和转化为,对每一个点计算它会在多少条边被割断之后成为子树的重心,大家可以看一看。Mr_Wu 的博客
CSP2019 树的重心 题解的更多相关文章
- [CSP-S2019]树的重心 题解
CSP-S2 2019 D2T3 考场上扔了T2来打这题的部分分,然后没看到数据范围是等号,不知道怎么判完全二叉树然后40分滚粗…… ---- 思路分析 很容易想到$O(n^2)$每次暴力找重心,这个 ...
- 题解-FJOI2014 树的重心
FJOI2014 树的重心 \(Q\) 组测试数据.给一棵树大小为 \(n\),求有多少个子树与其重心相同.重心可能有多个. 数据范围:\(1\le Q\le 50\),\(1\le n\le 200 ...
- CSP2019 Day2T3 树的重心
显然如果我们直接枚举断哪条边,剩下的两颗树的重心编号非常不好求,一个常见的想法是我们反过来,考虑每个节点有多少种情况作为树的重心,计算每个点对答案的贡献. 下面我们就需要大力分类讨论了.假设我们现在考 ...
- POJ 1655 求树的重心
POJ 1655 [题目链接]POJ 1655 [题目类型]求树的重心 &题意: 定义平衡数为去掉一个点其最大子树的结点个数,求给定树的最小平衡数和对应要删的点.其实就是求树的重心,找到一个点 ...
- 洛谷P3345 [ZJOI2015]幻想乡战略游戏(动态点分治,树的重心,二分查找,Tarjan-LCA,树上差分)
洛谷题目传送门 动态点分治小白,光是因为思路不清晰就耗费了不知道多少时间去gang这题,所以还是来理理思路吧. 一个树\(T\)里面\(\sum\limits_{v\in T} D_vdist(u,v ...
- 点分治模板(洛谷P4178 Tree)(树分治,树的重心,容斥原理)
推荐YCB的总结 推荐你谷ysn等巨佬的详细题解 大致流程-- dfs求出当前树的重心 对当前树内经过重心的路径统计答案(一条路径由两条由重心到其它点的子路径合并而成) 容斥减去不合法情况(两条子路径 ...
- POJ 1655 - Balancing Act - [DFS][树的重心]
链接:http://poj.org/problem?id=1655 Time Limit: 1000MS Memory Limit: 65536K Description Consider a tre ...
- BZOJ 3510 - 首都 「 $LCT$ 动态维护树的重心」
这题 FlashHu 的优化思路值得借鉴 前置引理 树中所有点到某个点的距离和中,到重心的距离和是最小的. 把两棵树通过某一点相连得到一颗新的树,新的树的重心必然在连接原来两棵树重心的路径上. 一棵树 ...
- poj 1655 树的重心
Balancing Act Time Limit: 1000MS Memory Limit: 65536K Total Submissions: 13178 Accepted: 5565 De ...
随机推荐
- Kubernetes容器集群管理环境 - Node节点的移除与加入
一.如何从Kubernetes集群中移除Node比如从集群中移除k8s-node03这个Node节点,做法如下: 1)先在master节点查看Node情况 [root@k8s-master01 ~]# ...
- MicroPython:基于TPYBoard集合MAX7219点阵模块制作表白女神神器
转载请注明文章来源,更多教程可自助参考docs.tpyboard.com,QQ技术交流群:157816561,公众号:MicroPython玩家汇 前言 又是一年毕业季,只有到了毕业季才会意识到自己又 ...
- Kubernetes DaemonSet(部署守护进程)
Kubernetes DaemonSet(部署守护进程) • 在每一个Node上运行一个Pod• 新加入的Node也同样会自动运行一个Pod 应用场景:Agent 官方文档:https://kuber ...
- JDBC与Druid简单介绍及Druid与MyBatis连接数据库
序言 java程序与数据建立连接,首先要从jdbc说起,然后直接上阿里认为宇宙最好的数据库连接池druid,然后再说上层程序对象与数据源映射关联关系的orm-mybatis. JDBC介绍 JDBC( ...
- 带你认识MySQL sys schema
前言: MySQL 5.7中引入了一个新的sys schema,sys是一个MySQL自带的系统库,在安装MySQL 5.7以后的版本,使用mysqld进行初始化时,会自动创建sys库. sys库里 ...
- 【UWP】实现 FindAncestor 绑定
在 WPF 里,我们是可以在 RelativeSource 上面实现的,举个例子: <Grid Tag="2"> <Button> <Grid Tag ...
- preventDefault, stopPropagation, return false -JS事件处理中的坑
我们以一个文件上传ui重设计为例子来探讨这几个函数的区别: 其中的html代码如下: <div class="file-upload"> <input type= ...
- Java编程基础——运算符和进制
Java编程基础——运算符和进制 摘要:本文主要介绍运算符和进制的基本知识. 说明 分类 Java语言支持如下运算符: ◆ 算术运算符:++,--,+,-,*,/,%. ◆ 赋值运算符:=,+=,-= ...
- cent OS 7 安装谷歌浏览器
我直接写一个shell 脚本, install_google.sh, bash 命令直接运行就好, 脚本内容如下: (切换root用户执行) set -e # 出错即退出 echo " ...
- 英语_金丝楠是紫楠(phoebeSheareri)的别名
姚黄魏紫俱凋零--红木家具今古谈(连载七) [上海木业网]楠木品种包括闽楠.细叶楠.红毛山楠.滇楠.白楠.紫楠.乌心楠.桢楠.水楠.香楠等二百余种之多,1997年的木材国家标准中就列入了八种.某些售卖 ...