CODEFORCES, 383C,Propagating tree

题意:有一棵以1为根节点共n个节点的的树,每一个节点上都一个value值,执行m次操作。

现在有2种操作:

1.查询某一个节点的当前值

2.给某一个节点加上一个val值,而每一个节点加上val,其儿子节点就需要-val

1 ≤ n, m ≤ 200000

策略:考虑到如果树是链形的 给根节点+val后,每一个点都要执行一次更新。最差会达到10^10级的复杂度。

所以在此基础上考虑成段维护。在每一次跟新操作,需要更新的是当前节点及其子树的值,所以我们可以用dfs序将树序转化为数组序,就可以知道每个节点对应的子树对应在数组中的范围。在更新时用线段树维护相应数组区域val值变化。

然后观察到,如果在奇数层+val,那么之后所有的奇数层+val,偶数层-val。偶数层+val同理。所以用2个线段树去维护奇数层节点和偶数层节点,即在维护奇数层的线段树中,偶数层节点的val值是没有意义的。每次在奇数层+val之后,我们就会去偶数层的线段树-val,反之亦然。然后每次查询奇数层节点去相应的线段树查找,偶数亦然。

但是如果更加仔细的观察发现,即我们如若要对奇数层+val 我们实际+(-val)值,对偶数层+val我们仍然+val这样来操作。然后在我们取奇数节点的时候 我们直接减去维护的值,即-val,而偶数层仍然+val,结果仍然正确。那么用一棵线段树我们就能做到维护了。

查询时将每层的val值自上而下求和后自身相加。

每次操作的时间复杂度都在logn

所以时间复杂度在mlog(n)即可满足要求

发现代码中lazytag可以并不用直接用sg_tree即可。浪费了一些空间。

<pre name="code" class="cpp">//Auhor:Wenjun Shi
//On 2016/7/14
#include <iostream>
#include <cstdio>
#include <vector>
#define MAX_N 200005
using namespace std;
int sg_tree[MAX_N*2],lazytag[MAX_N*2];//懒惰标记
bool niv[MAX_N];//深度奇偶
int in[MAX_N], out[MAX_N];//入出时间戳
int t;
int point[MAX_N];//dfs序中第i个数在线段树中的序号
int n,m,A[MAX_N];
vector<int> G[MAX_N];

void dfs(int u,int fa)
{
    // sq[++pos]=u;
    in[u]=++t;
    for(int i=0; i<G[u].size(); i++)
    {
        int v=G[u][i];
        if(v!=fa)
        {
            niv[v]=!niv[u];
            dfs(v,u);
        }
    }
    out[u]=t;
}

void add_edge(int u,int v)
{
    G[u].push_back(v);
    G[v].push_back(u);
}

void build_tree(int l,int r,int i)
{
    if(l==r) {point[l]=i;return;}
    int mid=(l+r)/2;
    build_tree(l,mid,i+i);
    build_tree(mid+1,r,i+i+1);
}

void add(int lt,int rt,int l,int r,int i,int val)
{
    if(rt<l||lt>r) return;
    if(lt<=l&&rt>=r)
    {
        lazytag[i]+=val;
        return ;
    }
    int mid=(l+r)/2;
    add(lt,rt,l,mid,i+i,val);
    add(lt,rt,mid+1,r,i+i+1,val);
}

int query(int q,int flag)
{
    int sum=0;
    while(q!=0)
    {
        sum+=flag*lazytag[q];
        q/=2;
    }
    return sum;
}

int main()
{
    int u,v,op,node,val;
    scanf("%d%d",&n,&m);
    for(int i=1; i<=n; i++) scanf("%d",&A[i]);
    for(int i=0; i<n-1; i++)
    {
        scanf("%d%d",&u,&v);
        add_edge(u,v);
    }
    t=0;
    dfs(1,0);
    build_tree(1,n,1);
    for(int i=0; i<m; i++)
    {
        scanf("%d%d",&op,&node);
        if(op==1)
        {
            scanf("%d",&val);
            add(in[node],out[node],1,n,1,(niv[node]&1)?-val:val);
        }
        if(op==2)
        {
           printf("%d\n",A[node]+query(point[in[node]],(niv[node]&1)?-1:1));
        }
    }
    return 0;
}


 

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值