CodeForces 383C

最新推荐文章于 2018-08-12 12:58:48 发布

原创最新推荐文章于 2018-08-12 12:58:48 发布 · 2.9k 阅读

0 ·

CC 4.0 BY-SA版权

数据结构-----树状数组专栏收录该内容

17 篇文章

订阅专栏

本文介绍了一种使用树状数组解决特定树形DP问题的方法。通过维护两个树状数组来分别记录奇数层与偶数层节点的值变化，从而实现对树形结构的有效更新与查询。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目大意：

有一颗树,两个操作：

1.x,val 将点x加上val，x的子节点，如果与x的距离为偶数加上val，否则减去val。

2.x 求点x的数值。

解决方法：

建立两个树状数组，分别代表每次修改对于奇数层与偶数层的影响，然后求解是奇数层在奇数树状数组中求，偶数层在偶数树状数组中求。

我的代码：

#include <cstdio>
#include <iostream>
#include <cstring>
#include <vector>
#include <algorithm>
#define maxn 500000
using namespace std;
int tree1[maxn],tree2[maxn];
vector <int> po[maxn];
int l[maxn],r[maxn];
int d[maxn];
int num[maxn];
int n,m,totl;
int dfs(int v,int fa,int dd){
  d[v]=dd;
  l[v]=totl++;
  for (int i=0;i<po[v].size();i++)
    if (po[v][i]!=fa){
      dfs(po[v][i],v,1-dd);
    }
  r[v]=totl-1;
  return 0;
}
int add1(int x,int v){
  //cout<<x<<" "<<v<<" *"<<endl;
  for (int i=x;i<=n;i+=i&(-i))
    tree1[i]+=v;
  return 0;
}
int add2(int x,int v){
  //cout<<x<<" "<<v<<" *"<<endl;
  for (int i=x;i<=n;i+=i&(-i))
    tree2[i]+=v;
  return 0;
}
int sum1(int x){
  int ans=0;
  for (int i=x;i>0;i-=i&(-i))
    ans+=tree1[i];
  return ans;
}
int sum2(int x){
  int ans=0;
  for (int i=x;i>0;i-=i&(-i))
    ans+=tree2[i];
  return ans;
}
int main (){
  //freopen("test.in","r",stdin);
  while (~scanf("%d%d",&n,&m)){
    memset(tree1,0,sizeof(tree1));
    memset(tree2,0,sizeof(tree2));
    for (int i=1;i<=n;i++)
        scanf("%d",&num[i]),po[i].clear();
    for (int i=1;i<=n-1;i++){
        int a,b;scanf("%d%d",&a,&b);
        po[a].push_back(b);
        po[b].push_back(a);
    }
    totl=1;dfs(1,0,0);
    for (int i=1;i<=m;i++){
        int po,v;scanf("%d%d",&po,&v);
        if (po==1){
            int val;scanf("%d",&val);
            // cout<<"val v "<<val<<" "<<v<<" "<<l[v]<<" "<<r[v]<<endl;
            if (d[v]){
                add1(l[v],val);
                add1(r[v]+1,-1*val);
                add2(l[v],-1*val);
                add2(r[v]+1,val);
            }
            else {
                add1(l[v],-1*val);
                add1(r[v]+1,val);
                add2(l[v],val);
                add2(r[v]+1,-1*val);
            }
        }
        else {
            if (d[v]) cout<<sum1(l[v])+num[v]<<endl;
            else cout<<sum2(l[v])+num[v]<<endl;
        }
    }
  }
  return 0;
}