牛客 —求和（dfs序+线段树）

题目链接https://ac.nowcoder.com/acm/contest/5158/I

思路

用dfs把一棵树转换成区间，然后题目就变成了用线段树来解决区间求和和单点修改的问题

代码

#include<iostream>
#include<string>
#include<map>
//#include<unordered_map>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#include<iomanip>
#include<cmath>
#include<fstream>
#define X first
#define Y second
#define best 131 
#define INF 0x3f3f3f3f
#define P pair<int,int>
#define ls p<<1
#define rs p<<1|1
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double eps=1e-5;
const double pai=acos(-1.0);
const int N=1e6+10;
const int maxn=110;
const int mod=1000000007;
int in[N],out[N],head[N],tot,cnt,id[N],a[N];
struct Node
{
    int to;
    int next;
}tr[N<<2];
struct node
{
    int l;
    int r;
    int sum;
}t[N<<2];
//vector<int>E[N];
void add(int x,int y)
{
    tr[tot].to=y;
    tr[tot].next=head[x];
    head[x]=tot++;
}
void dfs(int x,int y)
{
    in[x]=++cnt;
    id[cnt]=x;
//    for(int i=0;i<E[x].size();i++)
//  {
//        int end=E[x][i];
//        if(end==y) continue;
//        dfs(end,x);
//    }
    for(int i=head[x];i!=-1;i=tr[i].next)
    {
        int end=tr[i].to;
        if(end==y) continue;
        dfs(end,x);
    }
    out[x]=cnt;
}
void pushup(int p)
{
    t[p].sum=t[ls].sum+t[rs].sum;
}
void build(int p,int l,int r)
{
    t[p].l=l,t[p].r=r;
    if(l==r)
    {
        t[p].sum=a[id[l]];
        return ;
    }
    int mid=(l+r)>>1;
    bulid(ls,l,mid);
    bulid(rs,mid+1,r);
    pushup(p);
}
void change(int p,int pos,int v)
{
    if(t[p].l==t[p].r)
    {
        t[p].sum+=v;
        return ;
    }
    int mid=(t[p].l+t[p].r)>>1;
    if(pos<=mid) change(ls,pos,v);
    else change(rs,pos,v);
    pushup(p);
}
int ask(int p,int l,int r)
{
    if(l<=t[p].l&&t[p].r<=r)
    {
        return t[p].sum;
    }
    int mid=(t[p].l+t[p].r)>>1;
    int ans=0;
    if(l<=mid) ans+=ask(ls,l,r);
    if(r>mid) ans+=ask(rs,l,r);
    return ans; 
}
int main()
{
    int n,m,k;
    scanf("%d%d%d",&n,&m,&k);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    memset(head,-1,sizeof (head));
    for(int i=1;i<=n-1;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
//      E[x].push_back(y);
//      E[y].push_back(x);
        add(x,y);
        add(y,x);
    }
    dfs(k,0);
    build(1,1,n);
    while(m--)
    {
        int op,aa,x;
        scanf("%d%d",&op,&aa);
        if(op==1)
        {
            scanf("%d",&x);
            change(1,in[aa],x);
        }
        else
        {
            printf("%d\n",ask(1,in[aa],out[aa]));
        }
    }
    return 0;
}