A - Aragorn's Story 树剖

树上区间更新和查询,原理就是把树拆成几条链,dfs序是连续的,所以就把他们强加到线段树上,进行更新和查询。

进行两遍dfs,第一遍把深度,重儿子,父节点等一些信息找出,然后第二遍dfs时记录每个节点的链头,遍历时先走重儿子,把树上的每个节点按dfs序应到数组上,建树就行了。具体过程找个视频看看就懂了。写的不太规范,自己能看懂就好了。

#include<bits/stdc++.h>

using namespace std;
#define inf 0x3f3f3f3f
struct code
{
    int l,r,lazy;
} tree[50005*4];
vector<int>dap[50005];
int id_sum[50005],s,e,d;
int n,m,q,siz[50005],deep[50005],son[50005];
int sum[50005],top[50005],fa[50005],tri[50005];
void buid(int l,int r,int t)
{
    tree[t].l=l;
    tree[t].r=r;
    tree[t].lazy=0;
    if(l==r)
    {
        tree[t].lazy=id_sum[l];
        return ;
    }
    int mid=(l+r)/2;
    buid(l,mid,2*t);
    buid(mid+1,r,2*t+1);
}
void update(int l,int r,int t,int d)
{
    if(tree[t].l==l&&tree[t].r==r)
    {
        tree[t].lazy+=d;
        return ;
    }
    if(tree[t].lazy!=0)
    {
        tree[2*t].lazy+=tree[t].lazy;
        tree[2*t+1].lazy+=tree[t].lazy;
        tree[t].lazy=0;
    }
    int mid=(tree[t].l+tree[t].r)/2;
    if(r<=mid)
        update(l,r,2*t,d);
    else if(l>mid)
        update(l,r,2*t+1,d);
    else
    {
        update(l,mid,2*t,d);
        update(mid+1,r,2*t+1,d);
    }
}
int queuy(int t,int p)
{
    if(tree[t].l==tree[t].r)
        return tree[t].lazy;
    if(tree[t].lazy!=0)
    {
        tree[2*t].lazy+=tree[t].lazy;
        tree[2*t+1].lazy+=tree[t].lazy;
        tree[t].lazy=0;
    }
    int mid=(tree[t].l+tree[t].r)/2;
    if(p<=mid)
        return queuy(2*t,p);
    else
        return queuy(2*t+1,p);
}
void dfs(int x,int from)
{
    deep[x]=deep[from]+1;
    siz[x]=1;
    fa[x]=from;
    son[x]=0;
    int len=dap[x].size();
    for(int i=0; i<len; i++)
    {
        int ty=dap[x][i];
        if(ty==from)
            continue;
        dfs(ty,x);
        siz[x]+=siz[ty];
        if(siz[ty]>siz[son[x]])
            son[x]=ty;
    }
}
int cnt;
void qdfs(int x,int t)
{
    top[x]=t,tri[x]=cnt++;
    id_sum[cnt-1]=sum[x];
    if(son[x])qdfs(son[x],t);
    int len=dap[x].size();
    for(int i=0; i<len; i++)
    {
        int ty=dap[x][i];
        if(ty==son[x]||ty==fa[x])
            continue;
        qdfs(ty,ty);
    }
}
void init()
{
    cnt=1;
    deep[1]=0;
    dfs(1,1);
    qdfs(1,1);
    buid(1,n,1);
}
void Update(int s,int e,int d)
{
    int tx=top[s],ty=top[e];
    while(tx!=ty)
    {
        if(deep[tx]<deep[ty])
            swap(s,e),swap(tx,ty);
        update(tri[tx],tri[s],1,d);
        s=fa[tx],tx=top[s];
    }
    if(deep[s]<deep[e])
        swap(s,e);
    update(tri[e],tri[s],1,d);
}
char str[5];
int main()
{
    while(~scanf("%d%d%d",&n,&m,&q))
    {
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&sum[i]);
            dap[i].clear();
        }
        for(int i=1; i<=m; i++)
        {
            scanf("%d%d",&s,&e);
            dap[s].push_back(e);
            dap[e].push_back(s);
        }
        init();
        for(int i=0;i<q;i++)
        {
            scanf("%s",str);
            if(str[0]=='Q')
            {
                scanf("%d",&s);
                printf("%d\n",queuy(1,tri[s]));
            }
            else
            {
                scanf("%d%d%d",&s,&e,&d);
                if(str[0]=='I')
                    Update(s,e,d);
                else
                    Update(s,e,-d);
            }
        }
    }
    return 0;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值