CodeForces 276E Little Girl and Problem on Trees 树状数组

题意：一棵树只有一个顶点，然后由这个顶点引申出多条单链，对于输入 0 v x d，代表把距离V节点距离在d以内的所有节点增加x，对于输入 1 v,代表查询v节点的值；

解题：对根节点和根节点的所有孩子节点建一棵树状数组，树状数组下标为深度，用前缀和表示每个节点的值，在单个节点更新变化，即如在（2,5）区间加5，则位置2要+5，位置6要-5；

         然后要考虑是否跨根的情况，如果跨根就把影响直接更新到根的树状数组上，这样查询的时候就要同时查节点u的顶节点的树和根节点的树，同时更新时主要消除根节点的树和子节点的树的重复添加的情况。然后根节点的值就直接单独统计就好。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string.h>
#include<algorithm>
#include<vector>
using namespace std;
#define ls rt<<1
#define rs rt<<1|1
#define lowbit(x) (x&(-x))

const int maxn=600000;
const int maxm=600000;
struct node
{
    int u,v,next;
}edge[maxm];

//初始值都为0
struct BIT
{
    vector<int>b;
    int n;
    void init()
    {
        for(int i=1;i<=n;i++)
            b.push_back(0);
    }
    int getsum(int pos)
    {
        int tmp=0;
        for(int i=pos; i>0; i-=lowbit(i))
            tmp+=b[i-1];
        return tmp;
    }

    int update(int pos,int val)
    {
//        cout<<pos<<" update  "<<val<<endl;
        for(int i=pos; i<=n; i+=lowbit(i))
            b[i-1]+=val;//cout<<i<<" hou "<<b[i]<<endl;
    }
} B[maxn];

int em;
int head[maxn];

void init()
{
    em=0;
    memset(head,-1,sizeof(head));
}

void addedge(int u,int v)
{
    edge[em].u=u;
    edge[em].v=v;
    edge[em].next=head[u];
    head[u]=em++;
}

int dep[maxn];
int top[maxn];

void dfs(int x,int fa,int depth,int root)
{
    dep[x]=depth;
    top[x]=root;
    B[root].n=max(B[root].n,depth);
    B[1].n=max(B[1].n,depth);
    for(int i=head[x]; ~i; i=edge[i].next)
    {
        int v=edge[i].v;
        if(v==fa) continue;
        dfs(v,x,depth+1,root);
    }
}
int val[10];

void update(int v,int x,int d)
{
    int dis=dep[v];
    //bu guo geng
    int root=top[v];
    if(dis>d)
    {
        B[root].update(dep[v]-d,x);
        if(dep[v]+d+1<=B[root].n)B[root].update(dep[v]+d+1,-x);//cout<<root<<" root"<<endl;//??
    }
    else//guo geng
    {
        val[1]+=x;//geng xin geng的值
        B[root].update(1,x);//cout<<root<<" root"<<endl;
        if(dep[v]+d+1<=B[root].n) {B[root].update(dep[v]+d+1,-x);}//和总长度取min值
        B[1].update(1,x);//cout<<1<<" root"<<endl;
//        cout<<(dis-d+1)<<" "<<dis<<" "<<d<<" "<<B[1].n<<endl;
        if(d-dis+1<=B[1].n) B[1].update(d-dis+1,-x);//cout<<1<<" root"<<endl;
        //这里要消除自己的yingxiang
         B[root].update(1,-x);
         if(d-dis+1<=B[root].n) B[root].update(d-dis+1,+x);
    }
}

int query(int x)
{
    if(x==1) return val[1];
    int dis=dep[x];
    int tmp1=B[1].getsum(dis);

    int tmp2=B[top[x]].getsum(dis);
//    cout<<tmp1<<" query "<<tmp2<<endl;
    return tmp1+tmp2;
}

int op,u,v,x,d,n,q;

void work()
{
    dep[1]=0;
    for(int i=1;i<=n;i++)
        B[i].n=0;
    memset(top,0,sizeof(top));
    for(int i=head[1]; ~i; i=edge[i].next)
    {
        int v=edge[i].v;
        dfs(v,1,1,v);
    }
    for(int i=1;i<=n;i++)
        B[i].b.clear(),B[i].init();
//    for(int i=1;i<=n;i++)
//        cout<<B[i].n<<" "<<i<<endl;
    for(int i=1; i<=q; i++)
    {
        scanf("%d",&op);
        if(op==0)
        {
            scanf("%d%d%d",&v,&x,&d);//v d范围内加x
            update(v,x,d);
        }
        else
        {
            scanf("%d",&v);
            printf("%d\n",query(v));
        }
    }


}



//二维,树状数组维护变化,前缀和维护值


int main()
{
    while(scanf("%d%d",&n,&q)!=EOF)
    {
        init();
        val[1]=0;
        for(int i=1; i<=n-1; i++)
            scanf("%d%d",&u,&v),addedge(u,v),addedge(v,u);
        work();
//        cout<<"over"<<endl;
    }
    return 0;
}