CF324E Xenia and Tree
题目传送门
题意
有一颗树,初始\(1\)号点为红色,其余为蓝色,有两种操作:1.修改一个蓝点为红点。2.查询每个点最近红点的距离
题解
应该是较板题
最近根号写多了,首先想到的是: 每根号个询问重构一次,不超过根号就暴力枚举这个点和那根号个点的距离。
发现可能是对的,dp可以在\(O(n)\)复杂度处理出答案,每\(O(\sqrt{n})\) dp一次即可。
考虑需要\(O(1)\)LCA,于是学了一手。
如何dp?对于每个点,先dp它的儿子,然后他的答案就等于它儿子或父亲中答案最小的加1.
注意,要调用两遍这个dp。
你可以这样理解,这是一个混合dp,它实际上分两步进行:
第一步: 处理出每个节点向下最近的距离,这个可以从下往上dp一遍,
第二步:处理每个点向上经过它父亲的最近距离,由于根节点不能向上,所以答案不变,从上往下dp是对的。
我比较懒,写在一起, 调用两遍。
实现
#include
#include
#include
#include
using namespace std;
int read(){
int num=0, flag=1; char c=getchar();
while(!isdigit(c) && c!='-') c=getchar();
if(c == '-') c=getchar(), flag=-1;
while(isdigit(c)) num=num*10+c-'0', c=getchar();
return num*flag;
}
int min(int a, int b){return a p[N];
vector op;
void dfs(int x){
dfn[x] = ++tot, dep[x] = dep[fa[x]] + 1;
f[tot][0] = x;
for(auto i : p[x]){
if(i == fa[x]) continue;
fa[i] = x;
dfs(i);
f[++tot][0] = x;
}
}
int mindep(int x, int y){
return dep[x]>1] + 1;
for(int i=1; i r) swap(l, r);
return mindep(f[l][lg[r-l+1]], f[r-(1< sqr){
solve(1);
solve(1);
op.clear();
}
for(auto i : op){
ans[x] = min(ans[x], getDist(x, i));
}
printf("%d\n", ans[x]);
}
}
return 0;
}