题意:一棵树有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;
}