HDU 3966（树链剖分+点修改+点查询）

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

题目大意：营地的分布成树型。每个营地都有一些人，每次修改修改一条链上的所有营地的人数，每次查询单个点。

解题思路：

树链剖分基础题。

维护一个sum。

注意轻链修改时，点修改和边修改的不同。

由于树的结构与线段树点的顺序不太相同，因此需要做一个映射数组rank。故在线段树Build的时候，权值是camp[rank[l]]，rank这步的映射在dfs2的时候完成，rank[w[u]]=u;

Query单点u的时候, w[u]是其在线段树中的位置。

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include "cstdio"
#include "vector"
#include "cstring"
using namespace std;
#define maxn 50005
int camp[maxn],s[maxn],dep[maxn],pos[maxn],son[maxn],top[maxn],fa[maxn],w[maxn],rank[maxn],cnt,n;
vector<int> G[maxn];
void dfs1(int u,int pre,int d)
{
    s[u]=;fa[u]=pre;dep[u]=d;
    for(int i=;i<G[u].size();i++)
    {
        int v=G[u][i];
        if(v==pre) continue;
        dfs1(v,u,d+);
        s[u]+=s[v];
        if(son[u]!=-||s[v]>s[son[u]]) son[u]=v;
    }
}
void dfs2(int u,int tp)
{
    w[u]=++cnt;top[u]=tp;rank[w[u]]=u;
    if(son[u]==-) return;
    dfs2(son[u],tp);
    for(int i=;i<G[u].size();i++)
    {
        int v=G[u][i];
        if(v!=son[u]&&v!=fa[u])
            dfs2(v,v);
    }
}
//Segment-Tree
#define lson l,mid,root<<1
#define rson mid+1,r,root<<1|1
int col[maxn<<],sum[maxn<<];
void PushUp(int root)
{
    sum[root]=sum[root<<]+sum[root<<|];
}
void Build(int l,int r,int root)
{
    col[root]=;
    if(l==r)
    {
        sum[root]=camp[rank[l]];
        return;
    }
    int mid=(l+r)>>;
    Build(lson);
    Build(rson);
    PushUp(root);
}
void PushDown(int root,int m)
{
    if(col[root])
    {
        col[root<<]+=col[root];
        col[root<<|]+=col[root];
        sum[root<<]+=(m-(m>>))*col[root];
        sum[root<<|]+=(m>>)*col[root];
        col[root]=;
    }
}
void Update(int L,int R,int v,int l,int r,int root)
{
    if(L<=l&&r<=R)
    {
        col[root]+=v;
        sum[root]+=(r-l+)*v;
        return;
    }
    PushDown(root,r-l+);
    int mid=(l+r)>>;
    if(L<=mid) Update(L,R,v,lson);
    if(R>mid) Update(L,R,v,rson);
    PushUp(root);
}
int Query(int p,int l,int r,int root)
{
    if(l==r) return sum[root];
    PushDown(root,r-l+);
    int mid=(l+r)>>,ret=;
    if(p<=mid) ret=Query(p,lson);
    else ret=Query(p,rson);
    return ret;
}
void Change(int x,int y,int v)
{
    while(top[x]!=top[y])
    {
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        Update(w[top[x]],w[x],v,,n,);
        x=fa[top[x]];
    }
    if(dep[x]>dep[y]) swap(x,y);
    Update(w[x],w[y],v,,n,);//与边修改不同
}
int main()
{
    //freopen("in.txt","r",stdin);
    int m,p,u,v,c;
    while(scanf("%d%d%d",&n,&m,&p)!=EOF)
    {
        memset(son,-,sizeof(son));
        for(int i=;i<=n;i++) G[i].clear();
        cnt=;
        for(int i=;i<=n;i++)
            scanf("%d",&camp[i]);
        for(int i=;i<=m;i++)
        {
            scanf("%d%d",&u,&v);
            G[u].push_back(v);
            G[v].push_back(u);
        }
        dfs1(,,);
        dfs2(,);
        Build(,n,);
        char cmd[];
        for(int i=;i<=p;i++)
        {
            scanf("%s",&cmd);
            if(cmd[]=='I')
            {
                scanf("%d%d%d",&u,&v,&c);
                Change(u,v,c);
            }
            if(cmd[]=='D')
            {
                scanf("%d%d%d",&u,&v,&c);
                Change(u,v,-c);
            }
            if(cmd[]=='Q')
            {
                scanf("%d",&c);
                printf("%d\n",Query(w[c],,n,));
            }
        }
    }

}

11757337

2014-09-29 16:51:29

Accepted

3966

2484MS

8308K

3299 B

C++

Physcal