hdu3966（树链剖分）

update： 2018-01-27

下面的应该是很久之前抄的代码了，，当时应该是费了很大劲弄明白了一点吧=_=||

最近一直再刷数据结构，今天终于刷到树链剖分了，看了一下代码感觉挺清晰的...

具体讲解就看高级数据结构这本书吧，p401，感觉写的很清晰易懂。

最近几天最大的感受就是，半年前怎么都搞不明白的东西现在可以轻轻松松看懂了。我想可能就是当时这些东西确实超出了我的理解能力吧，其实后来这半年感觉并没有什么进步，应该就是不知不觉中提升了吧，忽然想起来半年前去看一年前的东西也是一样的感觉。

集训不剩几天了，之后还有别的事情要做呢，加油吧～

---

---

题目连接：http://acm.hdu.edu.cn/showproblem.php?pid=3966

参考：http://blog.csdn.net/acdreamers/article/details/10594121

明天再补代码，，睡觉！！

----------------------------------------------------------------------------------

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
const int maxn=50010;

struct edge
{
    int v,nex;
}e[maxn<<1];

int n,m,q;
int cur;
int w[maxn],siz[maxn],top[maxn],son[maxn];
 // 节点权重  子树节点数   链首        重儿子  
int dep[maxn],f[maxn],order[maxn],tid[maxn];
 // 深度        父亲    生成线段     遍历序列编号   

int head[maxn];
int cnt;

void init()
{
    memset(head,-1,sizeof(head));
    memset(son,-1,sizeof(son));
    cur=0;
    cnt=0;
}

void adde(int u,int v)
{
    e[cnt].v=v;
    e[cnt].nex=head[u];
    head[u]=cnt++;
}



//树链剖分
void dfs1(int u,int fa,int d)
{
    dep[u]=d;
    f[u]=fa;
    siz[u]=1;
    for(int i=head[u];i!=-1;i=e[i].nex)
    {
        int v=e[i].v;
        if(v!=fa)
        {
            dfs1(v,u,d+1);
            siz[u]+=siz[v];
            if(son[u]==-1||siz[v]>siz[son[u]])  //更新重儿子
                son[u]=v;
        }
    }
}

void dfs2(int u,int tp)
{
    top[u]=tp;
    tid[u]=++cur;
    order[tid[u]]=u;
    if(son[u]==-1) return ;
    dfs2(son[u],tp);
    for(int i=head[u];i!=-1;i=e[i].nex)
    {
        int v=e[i].v;
        if(v!=son[u]&&v!=f[u])
            dfs2(v,v);
    }
}

//线段树
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1

int sum[maxn<<2];
int add[maxn<<2];

void pushup(int rt)
{
    sum[rt]=max(sum[rt<<1],sum[rt<<1|1]);
}

void pushdown(int rt,int m)
{
    if(add[rt])
    {
        add[rt<<1]+=add[rt];
        add[rt<<1|1]+=add[rt];
        sum[rt<<1]+=add[rt]*(m-(m>>1));
        sum[rt<<1|1]+=add[rt]*(m>>1);
        add[rt]=0;
    }
}

void build(int l,int r,int rt)
{
    add[rt]=0;
    if(l==r)
    {
        sum[rt]=w[order[l]];
        return ;
    }
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(rt);
}
void update(int L,int R,int c,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        add[rt]+=c;
        sum[rt]+=c*(r-l+1);
        return ;
    }
    pushdown(rt,r-l+1);
    int m=(l+r)>>1;
    if(L<=m) update(L,R,c,lson);
    if(R>m) update(L,R,c,rson);
    pushup(rt);
}
int query(int pos,int l,int r,int rt)
{
    if(l==r) return sum[rt];
    pushdown(rt,r-l+1);
    int m=(l+r)>>1;
    int ans=0;
    if(pos<=m) ans=query(pos,lson);
    else ans=query(pos,rson);
    pushup(rt);
    return ans;
}

void change(int x,int y,int c)
{
    while(top[x]!=top[y])  //不在一条链
    {
        if(dep[top[x]]<dep[top[y]]) swap(x,y);  
        update(tid[top[x]],tid[x],c,1,n,1);   // 更新链首深度大的那一条链
        x=f[top[x]];  // x所在链已更新完，将x记为x与y连接的那个点
    }
    if(dep[x]>dep[y]) swap(x,y);
    update(tid[x],tid[y],c,1,n,1);
}

int main()
{
    char op[3];
    int a,b,c;
    int u,v;
    while(scanf("%d%d%d",&n,&m,&q)!=EOF)
    {
        init();
        for(int i=1;i<=n;i++)
            scanf("%d",&w[i]);
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d",&u,&v);
            adde(u,v);
            adde(v,u);
        }
        dfs1(1,0,0);
        dfs2(1,1);
        build(1,n,1);
        while(q--)
        {
            scanf("%s",op);
            if(op[0]=='Q')
            {
                scanf("%d",&a);
                printf("%d
",query(tid[a],1,n,1));
            }
            else
            {
                scanf("%d%d%d",&a,&b,&c);
                if(op[0]=='D') c=-c;
                change(a,b,c);
            }
        }
    }
    return 0;
}