hdu3966 Aragorn's Story 树链剖分

来源：互联网发布：webshell提权编辑：程序博客网时间：2024/04/27 20:47

题意：n个点构成的树，每个点都有一个权值。有三种操作： 1. I a b k 使a到b路径中所有点的点权增加k 2. D a b k 使a到b路径中的

所有点的点权减去k 3. Q c 查询点c的权值。

思路：树链剖分。剖分后，成段更新，单点查询，详见代码：

/*********************************************************  file name: hdu3966.cpp  author : kereo  create time:  2015年01月22日 星期四 08时40分25秒*********************************************************/#include<iostream>#include<cstdio>#include<cstring>#include<queue>#include<set>#include<map>#include<vector>#include<stack>#include<cmath>#include<string>#include<algorithm>using namespace std;typedef long long ll;const int sigma_size=26;const int N=50;const int MAXN=50000+50;const int inf=0x3fffffff;const double eps=1e-8;const int mod=100000000+7;#define L(x) (x<<1)#define R(x) (x<<1|1)#define PII pair<int, int>#define mk(x,y) make_pair((x),(y))int n,m,k,edge_cnt,cnt;char str[N];int head[MAXN],sz[MAXN],fa[MAXN],son[MAXN],dep[MAXN],top[MAXN],pos[MAXN],num[MAXN];struct Edge{    int v,next;}edge[MAXN<<1];struct node{    int l,r;    int val;}segtree[MAXN<<2];void init(){    edge_cnt=cnt=0;    memset(head,-1,sizeof(head));}void addedge(int u,int v){    edge[edge_cnt].v=v;    edge[edge_cnt].next=head[u]; head[u]=edge_cnt++;}void dfs1(int u,int pre,int depth){    sz[u]=1; dep[u]=depth; son[u]=0; fa[u]=pre;    for(int i=head[u];i!=-1;i=edge[i].next){        int v=edge[i].v;        if(v == pre)            continue;        dfs1(v,u,depth+1);        sz[u]+=sz[v];         if(sz[son[u]]<sz[v])            son[u]=v;    }}void dfs2(int u,int tp){    pos[u]=++cnt; top[u]=tp;    if(son[u]!=0)        dfs2(son[u],top[u]);    for(int i=head[u];i!=-1;i=edge[i].next){        int v=edge[i].v;        if(v == fa[u] || v == son[u])            continue;        dfs2(v,v);    }}void push_down(int rt){    if(segtree[rt].val){        segtree[L(rt)].val+=segtree[rt].val;        segtree[R(rt)].val+=segtree[rt].val;        segtree[rt].val=0;    }}void build(int rt,int l,int r){    segtree[rt].l=l; segtree[rt].r=r; segtree[rt].val=0;    if(l == r)        return ;    int mid=(l+r)>>1;    build(L(rt),l,mid); build(R(rt),mid+1,r);}void update(int rt,int l,int r,int add){    if(segtree[rt].l == l && segtree[rt].r == r){        segtree[rt].val+=add;        return ;    }    push_down(rt);    int mid=(segtree[rt].l+segtree[rt].r)>>1;    if(r<=mid)        update(L(rt),l,r,add);    else if(l>mid)        update(R(rt),l,r,add);    else{        update(L(rt),l,mid,add); update(R(rt),mid+1,r,add);    }}int query(int rt,int p){    if(segtree[rt].l == segtree[rt].r)        return segtree[rt].val;    push_down(rt);    int mid=(segtree[rt].l+segtree[rt].r)>>1;    if(p<=mid)        return query(L(rt),p);    else         return query(R(rt),p);}void solve(int u,int v,int add){    while(top[u]!=top[v]){        if(dep[top[u]]<dep[top[v]])            swap(u,v);        update(1,pos[top[u]],pos[u],add);        u=fa[top[u]];    }    if(dep[u]>dep[v])        swap(u,v);    update(1,pos[u],pos[v],add); //一开始当做边权了。。然后u!=v的时候才执行这句。。点权相同也要更新。。}int main(){    while(~scanf("%d%d%d",&n,&m,&k)){        init();        for(int i=1;i<=n;i++)            scanf("%d",&num[i]);        for(int i=0;i<m;i++){            int u,v;            scanf("%d%d",&u,&v);            addedge(u,v); addedge(v,u);        }        dfs1(1,-1,1); dfs2(1,1); build(1,1,n);        for(int i=1;i<=n;i++)            update(1,pos[i],pos[i],num[i]);        for(int i=0;i<k;i++){            scanf("%s",str);            int u,v,add;            if(str[0] == 'I'){                scanf("%d%d%d",&u,&v,&add);                solve(u,v,add);            }            if(str[0] == 'D'){                scanf("%d%d%d",&u,&v,&add);                solve(u,v,-add);            }            if(str[0] == 'Q'){                scanf("%d",&u);                printf("%d\n",query(1,pos[u]));            }          }    }return 0;}

0 0