hdu5029 Relief grain 点权树链剖分，线段树

最新推荐文章于 2018-08-31 10:53:30 发布

tzb592825420

最新推荐文章于 2018-08-31 10:53:30 发布

阅读量402

点赞数

分类专栏：树链剖分

本文链接：https://blog.csdn.net/u013044116/article/details/48394439

版权

树链剖分专栏收录该内容

4 篇文章 0 订阅

订阅专栏

点权树链剖分，求出所有更改区间，用一个类似差分前缀和的东西扫一遍。

点权与边权的剖分在查询上有一点小区别。

#pragma comment(linker,"/STACK:1024000000,1024000000")
#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<vector>
#include<set>
#include<map>
#include<queue>
#include<bitset>
#define fi first
#define se second
#define pii pair<int,int>
#define ll long long
#define inf 1<<30
#define eps 1e-8
using namespace std;
const int maxn=100005;
int son[maxn],top[maxn],dep[maxn],fa[maxn],siz[maxn],id[maxn],fid[maxn];
int sz;
vector<int>g[maxn];
int n,m;
vector<int>zh[maxn];
vector<int>fu[maxn];
int ans[maxn];
pii tree[maxn<<2];
int N=100000;
void dfs(int u,int f)
{
    siz[u]=1;
    son[u]=0;
    for(int i=0;i<g[u].size();i++) {
        if(g[u][i]==f)
            continue;
        int v=g[u][i];
        fa[v]=u;
        dep[v]=dep[u]+1;
        dfs(v,u);
        if(siz[v]>siz[son[u]]) son[u]=v;
        siz[u]+=siz[v];
    }
}
void build(int u,int w)
{
    id[u]=++sz;
    fid[sz]=u;
    top[u]=w;
    if(son[u]) build(son[u],w);
    for(int i=0;i<g[u].size();i++) {
        int v=g[u][i];
        if(v==fa[u]) continue;
        if(v!=son[u]) build(v,v);
    }
}
void T()
{
    int rt=(n+1)/2;
    sz=0;
    dep[rt]=0;
    fa[rt]=0;
    dfs(rt,0);
    build(rt,rt);
}
void solve(int a,int b,int c)
{
    int f1=top[a],f2=top[b];
    while(f1!=f2) {
        if(dep[f1]<dep[f2]) {
            swap(f1,f2);
            swap(a,b);
        }
        zh[id[f1]].push_back(c);
        fu[id[a]+1].push_back(c);
        a=fa[f1]; f1=top[a];
    }
    if(dep[a]>dep[b]) swap(a,b);
    zh[id[a]].push_back(c);
    fu[id[b]+1].push_back(c);
}
void pushUp(int rt)
{
    if(tree[rt<<1].fi>=tree[rt<<1|1].fi)
        tree[rt]=tree[rt<<1];
    else
        tree[rt]=tree[rt<<1|1];
}
void buildTree(int l,int r,int rt)
{
    if(l==r) {
        tree[rt].fi=0;
        tree[rt].se=0;
        return;
    }
    int m=(l+r)/2;
    buildTree(l,m,rt<<1);
    buildTree(m+1,r,rt<<1|1);
    pushUp(rt);
}
void update(int L,int a,int l,int r,int rt)
{
    if(L==l && L==r) {
        tree[rt].fi+=a;
        if(tree[rt].fi>0)
            tree[rt].se=l;
        else
            tree[rt].se=0;
        return;
    }
    int m=(l+r)/2;
    if(L<=m)
        update(L,a,l,m,rt<<1);
    else
        update(L,a,m+1,r,rt<<1|1);
    pushUp(rt);
}
int main()
{
    int a,b,c;
    while(~scanf("%d%d",&n,&m)) {
        if(n==0 && m==0) break;
        for(int i=1;i<=n;i++) {
            g[i].clear();
            zh[i].clear();
            fu[i].clear();
        }
        for(int i=0;i<n-1;i++) {
            scanf("%d%d",&a,&b);
            g[a].push_back(b);
            g[b].push_back(a);
        }
        T();
        for(int i=0;i<m;i++) {
            scanf("%d%d%d",&a,&b,&c);
            solve(a,b,c);
        }
        buildTree(1,N,1);
        for(int i=1;i<=sz;i++) {
            for(int j=0;j<zh[i].size();j++) {
                update(zh[i][j],1,1,N,1);
            }
            for(int j=0;j<fu[i].size();j++) {
                update(fu[i][j],-1,1,N,1);
            }
            ans[fid[i]]=tree[1].se;
        }
        for(int i=1;i<=n;i++)
            printf("%d\n",ans[i]);
    }
    return 0;
}