hdu3966 Aragorn's Story 树链剖分

标签：树链剖分

题意：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;
}

hdu3966 Aragorn's Story 树链剖分

标签：树链剖分

原文地址：http://blog.csdn.net/u011645923/article/details/43019181