code forces 383C(线段树)经典好题

题意: 两种操作,1.给出一个节点u,u的权值加上val,u的下一层孩子权值加上-val,下下层孩子加上val,一直

2.查询节点u的权值

思路:将树进行离散化,跑个dfs,然后节点和它的孩子节点都在一个连续的区间内。

        更新的时候,因为对于每一层的更新情况不一样,主要是层数奇偶性不同的更新情况不一样,我们可以建立两棵线段树,一个记录的是奇数层的节点的情况,一个是偶。当为奇数线段树的时候,不用管偶数层的节点。偶数线段树同理。

具体可以看一下代码:


#include<bits/stdc++.h>
using namespace std;
const int maxn = 2e5 + 10;
typedef long long ll;
#define clr(x,y) memset(x,y,sizeof x)
#define INF 0x3f3f3f3f


int a[maxn];
struct Edge{int to,next;}edge[maxn << 2];
int head[maxn],edge_num,num;
int dep[maxn],id[maxn],R[maxn],deps[maxn];
int p[maxn];
void Init()
{
    clr(head,-1);edge_num = 0;num = 0;
}
void add_edge(int x,int y)
{
    edge[edge_num] = (Edge){y,head[x]};head[x] = edge_num ++;
}
void dfs(int u,int pre,int d)
{
    dep[u] = d;id[u] = ++ num;p[num] = u;deps[num] = d;
    for(int i = head[u];i != -1;i = edge[i].next)
    {
        int v = edge[i].to;
        if(v == pre)continue;
        dfs(v,u,d + 1);
    }
    R[u] = num;
}
int add[maxn << 2][2];
void build(int l,int r,int rt)
{
    add[rt][0] = add[rt][1] = 0;
    if(l == r)return;
    int mid = (l + r) >> 1;
    build(l,mid,rt << 1);build(mid + 1,r,rt << 1|1);
}
void push_down(int rt)
{
    if(add[rt][0])
    {
        add[rt << 1][0] += add[rt][0];
        add[rt << 1|1][0] += add[rt][0];
        add[rt][0] = 0;
    }
    if(add[rt][1])
    {
        add[rt << 1][1] += add[rt][1];
        add[rt << 1|1][1] += add[rt][1];
        add[rt][1] = 0;
    }
}
void update(int L,int R,int x,int l,int r,int rt)
{
    if(L <= l && R >= r)
    {
        add[rt][0] -= x;add[rt][1] += x;return ;
    }
    push_down(rt);
    int mid = (l + r) >> 1;
    if(L <= mid)
        update(L,R,x,l,mid,rt <<1);
    if(R >= mid + 1)
        update(L,R,x,mid + 1,r,rt <<1|1);
}
int query(int pos,int l,int r,int rt)
{
    if(l == r)
    {
        return add[rt][deps[l]%2] + a[p[l]];
    }
    push_down(rt);
    int mid = (l + r) >> 1;
    if(pos <= mid)
        return query(pos,l,mid,rt << 1);
    else return query(pos,mid + 1,r,rt << 1|1);
}
int main()
{
    int n,m;
    while( ~ scanf("%d%d",&n,&m))
    {
        Init();
        for(int i = 1;i <= n;i ++)scanf("%d",&a[i]);
        for(int i = 1;i <= n - 1;i ++)
        {
            int x,y;scanf("%d%d",&x,&y);add_edge(x,y);add_edge(y,x);
        }
        dfs(1,1,1);
        build(1,n,1);

        while(m --)
        {
            int type;scanf("%d",&type);
            if(type == 1)
            {
                int x,y;scanf("%d%d",&x,&y);
                if(dep[x] % 2 == 0)y = -y;
                update(id[x],R[x],y,1,num,1);
            }
            else
            {
                int x;scanf("%d",&x);
                printf("%d\n",query(id[x],1,num,1));
            }
        }
    }
    return 0;
}


猜你喜欢

转载自blog.csdn.net/xiaolonggezte/article/details/79139993