Codeforces Round #169 (Div. 2) E.Little Girl and Problem on Trees(成段更新)

题意:一棵树有n个节点,但是除了根节点1外,其他节点都的出度和入度加起来就只有2(就是棵树是由几条链的第一个节点粘在一起的),现在有两种操作(1)"0 v x d",在距离v节点距离d以内的所有节点的权值都加上1,(2)"1 v",查询节点v上的权值。

        没想到办法静态建树,用vector<vector<LL> >去动态地建树。
        去掉根节点,对每条链建一个树状数组(表示这个链上的每个节点加了多少权值),然后另外加一个树状数组表示从根节点开始的,距离为1到n的节点加了多少权值,维护好这两部分树状数组就可以了。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <map>
using namespace std;
typedef long long LL;
const int N=1e5+5;
int dis[N],fa[N];
vector<int> adj[N];
vector<vector<LL> > line;

int lowbit(int x){return x&(-x);}
void updata(int st,int ed,int valu,int id)
{
    if(st>ed) return;
    int mx=(int)line[id].size();
    for(int i=ed+1;i<mx;i+=lowbit(i)) line[id][i]-=valu;
    for(int i=st;i<mx;i+=lowbit(i)) line[id][i]+=valu;
}
LL query(int ind,int id)
{
    LL sum=0;
    int mx=(int)line[id].size();
    for(int i=ind;i>0;i-=lowbit(i)) sum+=line[id][i];
    return sum;
}
void dfs(int u,int pre,int cnt,int id)
{
    dis[u]=cnt;     fa[u]=id;
    line[id].push_back(0);
    for(int i=0;i<(int)adj[u].size();i++)
    {
        int v=adj[u][i];
        if(v==pre) continue;
        dfs(v,u,cnt+1,id);
    }
}
int main()
{
    int n,q;
    scanf("%d%d",&n,&q);
    for(int i=0;i<n-1;i++)
    {
        int a,b; scanf("%d%d",&a,&b);
        adj[a].push_back(b);
        adj[b].push_back(a);
    }
    int tot=(int)adj[1].size();
    line.resize(tot+5);
    for(int i=0;i<(int)adj[1].size();i++)
    {
        line[i+1].push_back(0);
        dfs(adj[1][i],1,1,i+1);
    }

    line[0].resize(N,0);

    LL sum_root=0;
    while(q--)
    {
        int cmd; scanf("%d",&cmd);
        if(cmd==0)
        {
            int v,x,d; scanf("%d%d%d",&v,&x,&d);
            if(v==1)
            {
                sum_root+=x;
                updata(1,d,x,0);
            }
            else
            {
                int id=fa[v];
                if(d<dis[v]) updata(dis[v]-d,dis[v]+d,x,id);
                else
                {
                    sum_root+=x;
                    updata(1,dis[v]+d,x,id);
                    updata(1,d-dis[v],x,0);
                    updata(1,d-dis[v],-x,id);
                }
            }
        }
        else
        {
            int v; scanf("%d",&v);
            if(v==1) printf("%I64d\n",sum_root);
            else
            {
                int id=fa[v];
            //  cout<<query(dis[v],id)<<" "<<query(dis[v],0)<<endl;
                LL ans=query(dis[v],id)+query(dis[v],0);
                printf("%I64d\n",ans);
            }
        }
    }
    return 0;
}


猜你喜欢

转载自blog.csdn.net/shiqi_614/article/details/8737473