标签：ch int 题解 LCA P4556 text siz Vani typ
题意
一棵 \(n\) 个节点的树，\(m\) 次修改，给 \(u\) 到 \(v\) 路径上每一个点一件种类为 \(z\) 的物品，求最后每个节点个数最多的物品种类。
思路
将题意再精简一下就是维护树上单点染色众数，修改是一条链。
（1）树上差分
想到修改一条路径其实可以用树上差分的思想，将路径划分为两部分 \(u\rightarrow \text{LCA}(u,v)\) 以及 \(\text{LCA}(u,v)\rightarrow v\)，我们的差分操作是对 \(u\) 和 \(v\) 种类 \(z\) 的统计 \(+1\)，对 \(\text{LCA}(u,v)\) 以及 \(fa(\text{LCA}(u,v))\) 种类 \(z\) 的统计 \(-1\)。
（2）求 \(\text{LCA}\)
轻重链剖分然后跳 \(top(u)\) 即可。
（3）线段树维护
因为每次只有“染色”操作，染色众数不需要回滚到原来的状态，于是我们可以用权值线段树来维护，对于每一个单点都建一棵权值线段树（要先离散化）。
（4）线段树合并
可以发现，节点 \(u\) 的信息修改一定是源自其内部节点的一条链，于是我们可以向上合并线段树，就等价于求差分数组的前缀和，即最终答案了。
代码
点击查看代码
int n,m;
struct Question{
int x,y,z;
}q[maxn];
struct Graph{
struct edge{
int to,nxt;
}e[maxn<<1];
int head[maxn],cnt;
inline void add_edge(int u,int v){
e[++cnt].to=v,e[cnt].nxt=head[u],head[u]=cnt;
}
int fa[maxn],son[maxn],dep[maxn],siz[maxn],top[maxn];
inline void dfs1(int u,int f,int d){
dep[u]=d,fa[u]=f,siz[u]=1;
int maxson=-1;
for(int i=head[u];i;i=e[i].nxt){
int v=e[i].to;
if(v==f) continue;
dfs1(v,u,d+1);
siz[u]+=siz[v];
if(siz[v]>maxson){
maxson=siz[v],son[u]=v;
}
}
}
inline void dfs2(int u,int t){
top[u]=t;
if(!son[u]) return;
dfs2(son[u],t);
for(int i=head[u];i;i=e[i].nxt){
int v=e[i].to;
if(v==fa[u]||v==son[u]) continue;
dfs2(v,v);
}
}
inline int get_lca(int x,int y){
while(top[x]!=top[y]){
if(dep[top[x]]<dep[top[y]]) swap(x,y);
x=fa[top[x]];
}
if(dep[x]<dep[y]) return x;
else return y;
}
}G;
int rt[maxn],tot;
struct SegmentTree{
#define mid ((l+r)>>1)
int ch[maxm][2],siz[maxm],typ[maxm];
inline void push_up(int x){
if(siz[ch[x][0]]>=siz[ch[x][1]]){
siz[x]=siz[ch[x][0]],typ[x]=typ[ch[x][0]];
}
else{
siz[x]=siz[ch[x][1]],typ[x]=typ[ch[x][1]];
}
}
inline void insert(int &x,int l,int r,int p,int k){
if(!x) x=++tot;
if(l==r){
siz[x]+=k,typ[x]=l;
return;
}
if(p<=mid) insert(ch[x][0],l,mid,p,k);
else insert(ch[x][1],mid+1,r,p,k);
push_up(x);
}
inline void merge(int &x,int &y,int l,int r){
if(!x||!y){
x+=y;
return;
}
if(l==r){
siz[x]+=siz[y],typ[x]=l;
return;
}
merge(ch[x][0],ch[y][0],l,mid),merge(ch[x][1],ch[y][1],mid+1,r);
push_up(x);
}
}tree;
int maxx,ans[maxn];
vector<int> g;
inline void dfs(int u,int fa){
for(int i=G.head[u];i;i=G.e[i].nxt){
int v=G.e[i].to;
if(v==fa) continue;
dfs(v,u);
tree.merge(rt[u],rt[v],1,g.size());
}
if(tree.siz[rt[u]]){
ans[u]=g[tree.typ[rt[u]]-1];
}
}
int main(){
n=read(),m=read();
for(int i=1;i<n;i++){
int u=read(),v=read();
G.add_edge(u,v),G.add_edge(v,u);
}
G.dfs1(1,0,1);G.dfs2(1,1);
for(int i=1;i<=m;i++){
q[i].x=read(),q[i].y=read(),q[i].z=read();
g.push_back(q[i].z);
}
sort(g.begin(),g.end());
g.erase(unique(g.begin(),g.end()),g.end());
for(int i=1;i<=m;i++){
int l=G.get_lca(q[i].x,q[i].y);
int tmp=lower_bound(g.begin(),g.end(),q[i].z)-g.begin()+1;
tree.insert(rt[q[i].x],1,g.size(),tmp,1);
tree.insert(rt[q[i].y],1,g.size(),tmp,1);
tree.insert(rt[l],1,g.size(),tmp,-1);
if(G.fa[l]){
tree.insert(rt[G.fa[l]],1,g.size(),tmp,-1);
}
}
dfs(1,0);
for(int i=1;i<=n;i++){
printf("%d\n",ans[i]);
}
return 0;
}
标签：ch,int,题解,LCA,P4556,text,siz,Vani,typ 来源： https://www.cnblogs.com/SoyTony/p/15918454.html
本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享； 2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关； 3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关； 4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除； 5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。