动态点分治入 门 题

题解:

看起来动态点分治由于维护了一棵树高最多为 \(\log n\) 的点分树,每次修改操作的次数是 \(O(\log n)\),但是处理父子关系还是很难维护的。


这个题第一眼看上去(如果不带修的话)是有点分治的思路在里面的。但是每次询问的图都在改变,因此就有了动态点分治

由于我们需要找出树上最远的两个关灯的点,点的状态是动态的。而点分治每次都是在找重心,因此把每一次的重心分层,并两两“连边”,就形成了点分树。在点分树上的儿子所管辖的点数一定小于父亲所管辖的点数的一半,所以树高是 \(O(\log n)\)。

点分治的核心是在子树重心处统计过重心的路径,而重点是不能在重心处统计同一子树内的答案。

本题要我们找最长关灯点对,因此我们需要找每个点 \(x\) 作为重心时的分治子树内到当前点 \(x\) 距离,并合并两个不同的子树中点的信息。由于只需要查询最大值,所以我们用一个堆来维护每一个分治子树中的信息。

又因为分治子树上的点到分治重心 \(k\) 的距离与当前点的距离不是线性关系(点 \(x\) 和点 \(k\) 不一定相邻,此时 \(x\to k\) 路径上的点就没有方便计算的途径),所以这个信息是子树 \(k\) 内的点到点 \(x\) 的距离。记最大值为 \(mx_k\)。

然后对于 \(x\) ,任意的 \(\left<x,y\right>\in \text{点分树}\) ,可以更新答案为\(\max_{\left<x,i\right>\in \text{点分树},\left<x,j\right>\in \text{点分树},i\ne j} mx_i+mx_j\)。此时我们发现,由于 \(i\ne j\) ,所以只需要取最大的两个 \(mx_k\) 就可以了。而这个信息也是动态的,所以应该再开一个堆。

此时我们每个点维护了两个堆

  1. 大根堆 \(\{q_x\}​\) 维护以 \(x​\) 为根的分治子树中到 \(fa_x​\)(指 \(x​\) 在分治子树上的父亲)的距离。
  2. 大根堆 \(\{q'_x\}\) 维护点分树上 \(x\) 的每个儿子 \(k\) 的 \(\max\{q_k\}\)。

接下来考虑如何开关灯。

我们每次只修改了一个点,并且一个点的信息只可能在它点分树上的祖先节点出现,有 \(O(\log n)\) 个,我们在构造点分树的时候可以预处理出每个点 \(x\) 到它第 \(i\) 个父亲的距离,记作 \(d_{i,x}\),那个父亲记为 \(f_{i,x}\),特殊地,每个点的直接父亲记为 \(fa_x\)。

此时考虑每次修改点 \(x​\) 对第 \(i​\) 个父亲的影响,看到上面两个堆的意义,还需要分类讨论。

  • 当关灯时,\(u=f_{i,x}\) 子树中多了一个距离为 \(d_{i,x}\) 的关灯点。需要在 \(\{q_u\}\) 中插入 \(d_{i+1,x}\)。看是否 \(d_{i+1,x}\) 成为了 \(\{q_u\}\) 中最大的元素,如果是,则把 \(\{q'_{fa_u}\}\) 中原来的 \(u\) 答案删掉,更新为这个答案。
  • 当开灯时,\(u\) 的子树中少了一个距离为 \(d_{i,x}\) 的关灯点,则需要在 \(\{q_u\}\) 中删除相应的元素,如果删除了最大的,再拿此时最大的补上去。

这时需要统计答案了。发现答案是 \(\max_{i=1}^n\max\{q'_i\}\) ,仍然是类似的堆操作。至此我们整道题维护了3种堆,届时输出最后一种堆的堆顶即可。

当子树内只有一个或没有关灯点的时候贡献都是 \(0\),要输出 \(-1\) 的情况可以在外面判。一个点的时候还要存一下答案是否在最后一种堆中…因此边界情况会比较多。堆的删除是用懒惰删除法,@Dew 教了我一种神仙的结构体写法非常赞。

其他:注意 \(x,y\) 分别指父子的时候不要搞混了…

点分治+堆所以时间复杂度为 \(O((n+m)\log^2n)\)。

Code:

#include<cstdio>
#include<cstring>
#include<queue>
using std::priority_queue;
int read()
{
int x=0;
char ch=getchar();
while(ch<'0'||ch>'9')
ch=getchar();
while(ch>='0'&&ch<='9')
{
x=x*10+ch-'0';
ch=getchar();
}
return x;
}
struct edge
{
int n,nxt;
edge(int n,int nxt)
{
this->n=n;
this->nxt=nxt;
}
edge(){}
}e[200000];
int head[100100],ecnt=-1;
void add(int from,int to)
{
e[++ecnt]=edge(to,head[from]);
head[from]=ecnt;
e[++ecnt]=edge(from,head[to]);
head[to]=ecnt;
}
bool used[100100];
int fa[100100],f[100100],rt,tot=0;
int sz[100100];
void dfs(int x,int from)
{
sz[x]=1;
f[x]=0;
for(int i=head[x];~i;i=e[i].nxt)
if(e[i].n!=from&&!used[e[i].n])
{
dfs(e[i].n,x);
sz[x]+=sz[e[i].n];
f[x]=f[x]>sz[e[i].n]?f[x]:sz[e[i].n];
}
f[x]=f[x]>tot-sz[x]?f[x]:tot-sz[x];
rt=f[x]<f[rt]?x:rt;
} int d[18][100100],cnt[100100],dpt=1; void Dfs(int x,int from)
{
d[++cnt[x]][x]=dpt;
++dpt;
for(int i=head[x];~i;i=e[i].nxt)
if(e[i].n!=from&&!used[e[i].n])
Dfs(e[i].n,x);
--dpt;
}
void divide(int x,int from)//仅初始化块
{
rt=0;
tot=sz[x];
dfs(x,x);
used[x=rt]=1;
fa[x]=from;
for(int i=head[x];~i;i=e[i].nxt)
if(!used[e[i].n])
divide(e[i].n,x);
for(int i=head[x];~i;i=e[i].nxt)
if(!used[e[i].n])
Dfs(e[i].n,x);
used[x]=0;
} bool col[100100];
int sum=0; struct heap
{
priority_queue<int> q;
priority_queue<int> p;
void maintain()
{
while(!p.empty()&&p.top()==q.top())
{
p.pop();
q.pop();
}
}
inline void POP(int x)
{p.push(x);}
inline void PUSH(int x)
{q.push(x);}
int TOP()
{
maintain();
return q.top();
}
inline int sz()
{return (q.size()-p.size());}
}q[100100],q_[100100],Q;
//q表示来源于自己子树中的
//q_表示存它爹的
int ans[100100];
bool gg[100100];
void upd(int x)
{
if(q[x].sz()==1)
{
if(!gg[x])
Q.POP(ans[x]);
ans[x]=q[x].TOP();
gg[x]=1;
}
else if(!q[x].sz())
{
if(!gg[x])
Q.POP(ans[x]);
gg[x]=0;
ans[x]=0;
Q.PUSH(0);
}
else
{
if(!gg[x])
Q.POP(ans[x]);
gg[x]=0;
int g=q[x].TOP();
q[x].POP(g);
ans[x]=g+q[x].TOP();
q[x].PUSH(g);
Q.PUSH(ans[x]);
}
}
void change(int x)
{
int y=x,tmp=0;
if(col[x])
{
col[x]=0;
--sum;
while(fa[y])
{
++tmp;
//先考虑y对fa[y]的原贡献
if(q_[y].TOP()==d[tmp][x])
{
q_[y].POP(d[tmp][x]);
//要删除一些元素了
q[fa[y]].POP(d[tmp][x]);
if(q_[y].sz())
q[fa[y]].PUSH(q_[y].TOP());
upd(fa[y]);
}
else
q_[y].POP(d[tmp][x]);
y=fa[y];
}
q[x].POP(0);
upd(x);
}
else
{
col[x]=1;
++sum;
q[x].PUSH(0);
upd(x);
while(fa[y])
{
++tmp;
if(!q_[y].sz()||d[tmp][x]>q_[y].TOP())
{
if(q_[y].sz())
q[fa[y]].POP(q_[y].TOP());
q[fa[y]].PUSH(d[tmp][x]);
upd(fa[y]);
}
q_[y].PUSH(d[tmp][x]);
y=fa[y];
}
}
} int main()
{
memset(head,-1,sizeof(head)); f[0]=1e9;
int n,u,v;
n=read();
for(int i=1;i<n;++i)
{
u=read();
v=read();
add(u,v);
}
sz[1]=n;
divide(1,0);
for(int i=1;i<=n;++i)
Q.PUSH(0);
for(int i=1;i<=n;++i)
change(i);
char s[100];
int m;
m=read();
while(m--)
{
scanf("%s",s);
if(s[0]=='G')
{
if(!sum)
puts("-1");
else if(sum==1)
puts("0");
else
printf("%d\n",Q.TOP());
}
else
{
u=read();
change(u);
}
}
return 0;
}
05-20 21:18