CF276E Little Girl and Problem on Trees

一、题目

点此看题

二、解法

这道题翻译有点问题，有一个性质没有提到：除$1$以外的点，每个点的度数最多是$2$

对于一个修改，先考虑把当前点所在的这条链更新，对每条链的点重新编号即可。再考虑更新$1$连出去的其他链，这些链影响的深度都是一样的，所以再维护每一个深度的公共权值（注意要在原来的链删去这个权值）

思路不难，实则难写的一批，心态被写炸了。。。。

#include <cstdio>
#include <iostream>
using namespace std;
const int M = 100005;
int read()
{
	int x=0,f=1;char c;
	while((c=getchar())<'0' || c>'9') {if(c=='-') f=-1;}
	while(c>='0' && c<='9') {x=(x<<3)+(x<<1)+(c^48);c=getchar();}
	return x*f;
}
int n,m,cnt,tot,ans,f[M],dep[M],num[M],l[M],r[M];
int val[4*M][2];
struct edge
{
	int v,next;
}e[2*M];
void dfs(int u,int fa)
{
	num[u]=++cnt;
	dep[u]=dep[fa]+1;
	for(int i=f[u];i;i=e[i].next)
		if(e[i].v^fa)
			dfs(e[i].v,u);
}
void modify(int i,int l,int r,int L,int R,int v,int d)
{
	if(l>R || L>r) return ;
	if(L<=l && r<=R)
	{
		val[i][d]+=v;
		return ;
	}
	int mid=(l+r)>>1;
	modify(i<<1,l,mid,L,R,v,d);
	modify(i<<1|1,mid+1,r,L,R,v,d);
}
void ask(int i,int l,int r,int id,int d)
{
	ans+=val[i][d];
	if(l==r) return ;
	int mid=(l+r)>>1;
	if(id<=mid) ask(i<<1,l,mid,id,d);
	else ask(i<<1|1,mid+1,r,id,d);
}
int main()
{
	n=read();m=read();
	for(int i=1;i<n;i++)
	{
		int u=read(),v=read();
		e[++tot]=edge{v,f[u]},f[u]=tot;
		e[++tot]=edge{u,f[v]},f[v]=tot;
	}
	num[1]=++cnt;dep[1]=1;
	for(int i=f[1];i;i=e[i].next)
	{
		int st=cnt+1;
		dfs(e[i].v,1);
		for(int j=st;j<=cnt;j++)
			l[j]=st,r[j]=cnt;
	}
	for(int i=1;i<=m;i++)
	{
		int op=read(),v=read();
		if(op==0)
		{
			int x=read(),d=read();
			if(v!=1) modify(1,1,n,max(l[num[v]],num[v]-d),min(r[num[v]],num[v]+d),x,0);
			if(dep[v]<=d+1)
			{
				modify(1,1,n,1,d-dep[v]+2,x,1);
				if(dep[v]<=d) modify(1,1,n,l[num[v]],min(r[num[v]],l[num[v]]+d-dep[v]),-x,0);
			}
		}
		else
		{
			ans=0;
			ask(1,1,n,num[v],0);
			ask(1,1,n,dep[v],1);
			printf("%d\n",ans);
		}
	}
}