HDU 3966 Aragorn's Story（树链剖分+线段树）

HDU3966

三种操作：路径上所有点的人数+v;路径上所有点的人数-v;求某个节点x的人数

树链剖分的点权操作，以及单点查询。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<algorithm>
#define memset(a,v)  memset(a,v,sizeof(a))
#define max(a,b)   (a>b?a:b)
#define min(a,b)   (a<b?a:b)
#define swap(a,b)  (a=a+b,b=a-b,a=a-b)
#define eps 1.0E-8
using namespace std;
const int MAXL(5*1e4);
const int INF(0x7f7f7f7f);
const int mod(1e9+7);
typedef long long int LL;
struct node
{
    int to,next;
    node(){}
    node(int to,int next):to(to),next(next){}
}edge[2*MAXL+50];
int head[MAXL+50];
int top[MAXL+50];
int father[MAXL+50];
int depth[MAXL+50];
int siz[MAXL+50];
int son[MAXL+50];
int pos[MAXL+50];
int repos[MAXL+50];
int a[MAXL+50];
int tot,dpos,cnt;
int n,m,q;

void init()
{
    cnt=0,dpos=1;
    memset(head,-1);
    memset(son,-1);
}
inline void dfs1(int u,int fa,int dep)
{
    depth[u]=dep;father[u]=fa;siz[u]=1;
    for(int i=head[u];~i;i=edge[i].next)
    {
        int v=edge[i].to;
        if(v!=fa)
        {
            dfs1(v,u,dep+1);
            siz[u]+=siz[v];
            if(son[u]==-1||siz[v]>siz[son[u]])
                son[u]=v;
        }
    }
}

inline void dfs2(int u,int t)
{
    top[u]=t;
    pos[u]=dpos;
    repos[dpos]=u;
    dpos++;
    if(son[u]==-1)
        return;
    dfs2(son[u],t);
    for(int i=head[u];~i;i=edge[i].next)
    {
        int v=edge[i].to;
        if(v!=son[u]&&v!=father[u])
            dfs2(v,v);
    }
}

void add_edge(int u,int v)
{
    edge[cnt]=node(v,head[u]);
    head[u]=cnt++;
}

struct Node
{
    int l,r,data,lazy;
}tree[4*MAXL+50];

void PushUp(int k)
{
    tree[k].data=tree[k<<1].data+tree[k<<1|1].data;
}

void PushDown(int k)
{
    if(!tree[k].lazy) return;
    tree[k<<1].lazy+=tree[k].lazy;
    tree[k<<1|1].lazy+=tree[k].lazy;
    tree[k<<1].data+=tree[k].lazy*(tree[k<<1].r-tree[k<<1].l+1);
    tree[k<<1|1].data+=tree[k].lazy*(tree[k<<1|1].r-tree[k<<1|1].l+1);
    tree[k].lazy=0;
}

void build(int l,int r,int k)
{
    tree[k].l=l,tree[k].r=r,tree[k].lazy=0;
    if(l==r)
    {
        tree[k].data=a[repos[l]];
        return;
    }
    int mid=(l+r)>>1;
    build(l,mid,k<<1);
    build(mid+1,r,k<<1|1);
    PushUp(k);
}

void update(int ul,int ur,int v,int k)
{
    if(tree[k].l>=ul&&tree[k].r<=ur)
    {
        tree[k].data+=(tree[k].r-tree[k].l+1)*v;
        tree[k].lazy+=v;
        return;
    }
    PushDown(k);
    int mid=(tree[k].l+tree[k].r)>>1;
    if(ul<=mid)
        update(ul,ur,v,k<<1);
    if(ur>mid)
        update(ul,ur,v,k<<1|1);
    PushUp(k);
}

void change(int u,int v,int val)
{
    int fu=top[u],fv=top[v];
    int ans=0;
    while(fu!=fv)
    {
        if(depth[fu]<depth[fv])
            swap(fu,fv),swap(u,v);
        update(pos[fu],pos[u],val,1);
        u=father[fu];
        fu=top[u];
    }
    if(depth[u]>depth[v]) swap(u,v);
    update(pos[u],pos[v],val,1);
}

int ans;

void query(int qpoint,int k)
{
    if(tree[k].l==tree[k].r)
    {
        ans=tree[k].data;
        return;
    }
    PushDown(k);
    int mid=(tree[k].l+tree[k].r)>>1;
    if(qpoint<=mid)
        query(qpoint,k<<1);
    else
        query(qpoint,k<<1|1);
}


int main()
{
    while(~scanf("%d%d%d",&n,&m,&q))
    {
        init();
        for(int i=1;i<=n;i++)
            scanf("%d",a+i);
        for(int i=1;i<=m;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            add_edge(u,v);
            add_edge(v,u);
        }
        dfs1(1,0,0);
        dfs2(1,1);
        build(1,n,1);
        while(q--)
        {
            char ch[2];
            scanf("%s",ch);
            if(ch[0]=='Q')
            {
                int x;
                scanf("%d",&x);
                query(pos[x],1);
                printf("%d\n",ans);
            }
            if(ch[0]=='I')
            {
                int x,y,v;
                scanf("%d%d%d",&x,&y,&v);
                change(x,y,v);
            }
            if(ch[0]=='D')
            {
                int x,y,v;
                scanf("%d%d%d",&x,&y,&v);
                v=-v;
                change(x,y,v);
            }
        }
    }
}