Infiniti

   :: 首页  :: 新随笔  ::  ::  :: 管理

HDU - 3966 

思路 :树链剖分就是可以把一个路径上的点映射成几段连续的区间上。这样对于连续的区间可以用线段树维护,

对于每一段连续的区间都可以通过top [ ]数组很快的找到这段连续区间的头。跳的过程类似于 lca ,但这里 要注意的是

每一个点只属于一条链。 (重载运算符时要注意   node 一个 新的结点,不要 乱用*this 指针)

#include<bits/stdc++.h>
using namespace std;
#define MID int m = (l+r)/2
#define maxn 56789
#define inf 0x3f3f3f3f
struct node
{
    int sum,lazy,cnt;
    node()
    {
        sum=lazy=cnt=0;
    }
    node operator+(const node &a)const
    {
        node ret;
        ret.sum=a.sum+sum;
        ret.cnt=a.cnt+cnt;
        return ret;
    }
} tree[maxn*4];
char str[12];
vector<int>edge[maxn];
int data[maxn],n,m,id[maxn],fa[maxn],u,ans;
int son[maxn],top[maxn],tid[maxn],cnt,v,ad;
int deep[maxn],siz[maxn],id_data[maxn],q;
void pushdown(int root)
{
    if(tree[root].lazy==0)return ;
    tree[root*2].lazy+=tree[root].lazy;
    tree[root*2+1].lazy+=tree[root].lazy;
    tree[root*2].sum+=tree[root*2].cnt*tree[root].lazy;
    tree[root*2+1].sum+=tree[root*2+1].cnt*tree[root].lazy;
    tree[root].lazy=0;
}
void bulid(int root,int l,int r)
{
    tree[root].lazy=0;
    if(l==r)
    {
        tree[root].sum=id_data[l];
        tree[root].cnt=1;
        return ;
    }
    MID;
    bulid(root*2,l,m);
    bulid(root*2+1,m+1,r);
    tree[root]=tree[root*2]+tree[root*2+1];
}
void updata(int root,int l,int r,int L,int R,int ad)
{
    if(r<L||l>R)return;
    if(L<=l&&r<=R)
    {
        tree[root].sum+=tree[root].cnt*ad;
        tree[root].lazy+=ad;
        return ;
    }
    pushdown(root);
    MID;
    updata(root*2,l,m,L,R,ad);
    updata(root*2+1,m+1,r,L,R,ad);
    tree[root]=tree[root*2]+tree[root*2+1];
}
void query(int root,int l,int r,int L,int R)
{
    if(r<L||l>R)return ;
    if(L<=l&&r<=R)
    {
        ans+=tree[root].sum;
        return;
    }
    pushdown(root);
    MID;
    query(root*2,l,m,L,R);
    query(root*2+1,m+1,r,L,R);
}
void dfs1(int u,int pre,int ide)
{
    son[u]=-1,siz[u]=1;
    deep[u]=ide,fa[u]=pre;
    for(int i=0; i<edge[u].size(); i++)
    {
        int v=edge[u][i];
        if(v==pre)continue;
        dfs1(v,u,ide+1);
        if(son[u]==-1||siz[son[u]]<siz[v])
            son[u]=v;
    }
}
void dfs2(int u,int tp)
{
    top[u]=tp, tid[u]=++cnt;
    id_data[cnt]=data[u];
    if(son[u]!=-1)dfs2(son[u],tp);
    for(int i=0; i<edge[u].size(); i++)
    {
        int v=edge[u][i];
        if(v==fa[u]||v==son[u])continue;
        dfs2(v,v);
    }
}
void solve(int x,int y,int ad)
{
    int tx=top[x],ty=top[y];
    while(tx!=ty)
    {
        if(deep[tx]<deep[ty])swap(x,y),swap(tx,ty);
        updata(1,1,n,tid[tx],tid[x],ad);
        x=fa[tx],tx=top[x];
    }
    if(deep[x]<deep[y])swap(x,y);
    updata(1,1,n,tid[y],tid[x],ad);
}
int main()
{
    while(~scanf("%d%d%d",&n,&m,&q))
    {
        cnt=0;
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&data[i]);
            edge[i].clear();
        }
        while(m--)
        {
            scanf("%d%d",&u,&v);
            edge[u].push_back(v);
            edge[v].push_back(u);
        }
        dfs1(1,0,1);
        dfs2(1,1);
        bulid(1,1,n);
        while(q--)
        {
            scanf("%s",str);
            if(str[0]=='I')
            {
                scanf("%d%d%d",&u,&v,&ad);
                solve(u,v,ad);
            }
            else if(str[0]=='D')
            {
                scanf("%d%d%d",&u,&v,&ad);
                solve(u,v,-ad);
            }
            else
            {
                scanf("%d",&u);
                ans=0;
                query(1,1,n,tid[u],tid[u]);
                printf("%d\n",ans);
            }
        }
    }
    return 0;
}

  

posted on 2019-01-17 14:29  自由缚  阅读(144)  评论(0编辑  收藏  举报