CF 276E Little Girl and Problem on Trees

题目链接：http://codeforces.com/problemset/problem/276/E

题目大意：

一棵树有n个点，编号1~n，1为根，除根意外的所有结点，至多一个子结点。

有两种操作

0 v x d：和v结点距离不大于d的所有结点权值+x

1 v    ：输出结点v的权值

题目思路：

看似像区间更新+单点询问，但是却又不一样，因为是在一棵树上而不是线段上，但是我们发现给出的树是从根到叶子是一条线段，因为每个结点（除根结点）至多一个子节点。

所以我们可以看成是多条线段的区间更新+单点询问。

但是这并不能根本解决问题，因为可能需要更新所有线段，而且容易知道只会更新一条线段或者全部线段。

所以再加一条特殊的线段，用于在更新所有线段时用。

而区间更新+单点询问可以把树状数组逆运用。

代码：

#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <ctype.h>
#include <math.h>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <string>
#include <iostream>
#include <algorithm>
using namespace std;

//#define ll __int64
#define ls rt<<1
#define rs ls|1
#define lson l,mid,ls
#define rson mid+1,r,rs
#define middle (l+r)>>1
#define eps (1e-8)
#define clr_all(x,c) memset(x,c,sizeof(x))
#define clr(x,c,n) memset(x,c,sizeof(x[0])*(n+1))
#define MOD 1000000007
#define INF 0x3f3f3f3f
#define PI (acos(-1.0))
#define _mod(x,y) ((x)>0? (x)%(y):((x)%(y)+(y))%(y))
#define _abs(x) ((x)<0? (-(x)):(x))
#define getmin(x,y) (x= ((x)<0 || (y)<(x))? (y):(x))
#define getmax(x,y) (x= ((y)>(x))? (y):(x))
template <class T> void _swap(T &x,T &y){T t=x;x=y;y=t;}
template <class T> T _max(T x,T y){return x>y? x:y;}
template <class T> T _min(T x,T y){return x<y? x:y;}
int TS,cas=1;
const int M=100000+5;
int n,q,dis[M];
vector<int>init[M];
int tot;
int* val[M];
int nn[M],mp[M];


void bfs(int u){
	int i;
	queue<int>q;
	q.push(u);
	clr(dis,0,n+1);
	dis[u]=nn[tot=0]=1;
	while(!q.empty()){
		u=q.front();
		q.pop();
		for(i=0;i<init[u].size();i++){
			int& v=init[u][i];
			if(dis[v]) continue;
			nn[0]=_max(nn[0],dis[v]=dis[u]+1);
			if(dis[v] == 2) nn[mp[v]=++tot]=2;
			else if(dis[v] > 2) nn[mp[v]=mp[u]]++;
			q.push(v);
		}
	}
}

int lowbit(int x){return x&(-x);}

void update(int id,int p,int c){
	for(;p>0;p-=lowbit(p)) val[id][p]+=c;
}

int query(int id,int p){
	int ret=0;
	for(;p<=nn[id];p+=lowbit(p)) ret+=val[id][p];
	return ret;
}

void run(){
	int i,j;
	int u,op,v,x,d;
	for(i=1;i<=n;i++) init[i].clear();
	for(i=1;i<n;i++){
		scanf("%d%d",&u,&v);
		init[u].push_back(v);
		init[v].push_back(u);
	}
	bfs(1);
	for(i=0;i<=tot;i++)
		val[i]=(int*)calloc(nn[i]+1,sizeof(int));
	while(q--){
		scanf("%d",&op);
		if(op==0){
			scanf("%d%d%d",&v,&x,&d);
			if(v==1){
				update(0,_min(nn[0],d+1),x);
			}else{
				int id=mp[v],l=dis[v]-d,r=_min(nn[id],dis[v]+d);
				if(l > 1){
					update(id,r,x);
					update(id,l-1,-x);
				}else{
					int r0=_min(nn[0],2-l);
					update(0,r0,x);
					if(r0 < r){
						update(id,r,x);
						update(id,r0,-x);
					}
				}
			}	
		}else{
			scanf("%d",&v);
			int ret=query(0,dis[v]);
			if(v!=1) ret+=query(mp[v],dis[v]);
			printf("%d\n",ret);
		}
	}
	for(i=0;i<=tot;i++) free(val[i]);
}

void preSof(){
}

int main(){
    //freopen("input.txt","r",stdin);
    //freopen("output.txt","w",stdout);
    preSof();
    //run();
    while((~scanf("%d%d",&n,&q))) run();
    //for(scanf("%d",&TS);cas<=TS;cas++) run();
    return 0;
}