​点击打开链接​

题目要求在u到v这条链上加入w这个元素 并统计每个点上最多的元素是谁 只要在链首加入一个w 链尾加入一个-w即可 有点离线处理的意思

树链剖分中 一条链会被划分成logn条小链 这些小链在树剖版dfs序中是连续的 所以要分段更新

#include <bits/stdc++.h>
using namespace std;

struct node1
{
int v;
int next;
};

struct node2
{
int l;
int r;
int val;
int id;
};

vector <int> pre[100010];
node1 edge[200010];
node2 tree[400010];
int first[100010],fa[100010],deep[100010],sum[100010],son[100010],top[100010],mp1[100010],mp2[100010],ans[100010];
int n,q,num;

void addedge(int u,int v)
{
edge[num].v=v;
edge[num].next=first[u];
first[u]=num++;
}

void dfsI(int cur)
{
int i,v;
sum[cur]=1,son[cur]=-1;
for(i=first[cur];i!=-1;i=edge[i].next)
{
v=edge[i].v;
if(v!=fa[cur])
{
fa[v]=cur,deep[v]=deep[cur]+1;
dfsI(v);
sum[cur]+=sum[v];
if(son[cur]==-1||sum[son[cur]]<sum[v])
{
son[cur]=v;
}
}
}
}

void dfsII(int cur,int tp)
{
int i,v;
num++;
top[cur]=tp,mp1[cur]=num,mp2[num]=cur;
if(son[cur]==-1) return;
dfsII(son[cur],tp);
for(i=first[cur];i!=-1;i=edge[i].next)
{
v=edge[i].v;
if(v!=fa[cur]&&v!=son[cur])
{
dfsII(v,v);
}
}
}

void solve(int u,int v,int w)
{
while(top[u]!=top[v])
{
if(deep[top[u]]<deep[top[v]]) swap(u,v);
pre[mp1[top[u]]].push_back(w);
pre[mp1[u]+1].push_back(-w);
u=fa[top[u]];
}
if(deep[u]<deep[v]) swap(u,v);
pre[mp1[v]].push_back(w);
pre[mp1[u]+1].push_back(-w);
}

void pushup(int cur)
{
if(tree[2*cur].val>tree[2*cur+1].val)
{
tree[cur].val=tree[2*cur].val;
tree[cur].id=tree[2*cur].id;
}
else if(tree[2*cur].val==tree[2*cur+1].val)
{
tree[cur].val=tree[2*cur].val;
tree[cur].id=min(tree[2*cur].id,tree[2*cur+1].id);
}
else
{
tree[cur].val=tree[2*cur+1].val;
tree[cur].id=tree[2*cur+1].id;
}
}

void build(int l,int r,int cur)
{
int m;
tree[cur].l=l;
tree[cur].r=r;
tree[cur].val=0;
tree[cur].id=0;
if(l==r) return;
m=(l+r)/2;
build(l,m,2*cur);
build(m+1,r,2*cur+1);
}

void update(int tar,int val,int cur)
{
if(tree[cur].l==tree[cur].r)
{
tree[cur].val+=val;
if(tree[cur].val==0) tree[cur].id=0;
else tree[cur].id=tree[cur].l;
return;
}
if(tar<=tree[2*cur].r) update(tar,val,2*cur);
else update(tar,val,2*cur+1);
pushup(cur);
}

int main()
{
int i,j,u,v,w,tar,maxx;
while(scanf("%d%d",&n,&q)!=EOF)
{
if(n==0&&q==0) break;
memset(first,-1,sizeof(first));
num=0;
for(i=1;i<=n-1;i++)
{
scanf("%d%d",&u,&v);
addedge(u,v);
addedge(v,u);
}
fa[1]=-1,deep[1]=1;
dfsI(1);
num=0;
dfsII(1,1);
for(i=1;i<=n;i++) pre[i].clear();
maxx=0;
while(q--)
{
scanf("%d%d%d",&u,&v,&w);
maxx=max(maxx,w);
solve(u,v,w);
}
build(1,maxx,1);
for(i=1;i<=n;i++)
{
for(j=0;j<pre[i].size();j++)
{
tar=pre[i][j];
if(tar>0) update(tar,1,1);
else update(-tar,-1,1);
}
ans[mp2[i]]=tree[1].id;
}
for(i=1;i<=n;i++)
{
printf("%d\n",ans[i]);
}
}
return 0;
}