1. 程式人生 > >【樹鏈剖分】樹上的詢問

【樹鏈剖分】樹上的詢問

                                            【題目描述】

給你一棵具有N個點(編號為1到N)M條邊的樹,並給定各個點權的值,然後有3種操作: I C1 C2 K:把C1與C2的路徑上的所有點權值加上K D C1 C2 K:把C1與C2的路徑上的所有點權值減去K Q C:查詢節點編號為C的權值

【輸入格式】

輸入包含多組測試資料。 對於每組測試資料的第一行包含三個整數N,M,P,分別表示N個結點(1<=N<=50000),M條邊(M=N-1)和P(1≤P≤100000)個操作。 接下來一行為N個整數A1,A2,...AN(0≤Ai≤1000),表示一開始每個結點的值。 接下來的M行,每行兩個整數u和v,表示u到v有一條邊。 接下來P行,表示相應的操作,其中0<=k<=1000。

【輸出格式】

對於每個詢問輸出一行,為詢問結點的值。

樹鏈剖分的裸題,不過對區間的操作比較簡單,我們可以直接用樹狀陣列的區間修改與單點查詢,沒必要寫線段樹,程式碼上會簡短不少

#include<iostream>
#include<cstdio>
#include<queue>
#include<iomanip>
#include<cstring>
#include<algorithm>
#include<bitset>
using namespace std;
int n,m,q,a[50005],h[50005],cnt,size[50005],prt[50005],son[50005],deep[50005],seg[50005],top[50005],rev[50005],tree[50005];
struct edge
{
	int to,next;
}w[100005];
void add(int x,int y)
{
	cnt++;
	w[cnt].to=y;
	w[cnt].next=h[x];
	h[x]=cnt;
}
void dfs(int x,int fa)
{
	prt[x]=fa;
	size[x]=1;
	deep[x]=deep[fa]+1;
	for(int i=h[x];i;i=w[i].next)
	{
		int to=w[i].to;
		if(to==fa) continue;
		dfs(to,x);
		size[x]+=size[to];
		if(size[to]>size[son[x]]) son[x]=to;
	}
}
void dfs2(int x)
{
	if(son[x])
	{
		top[son[x]]=top[x];
		seg[son[x]]=++seg[0];
		rev[seg[0]]=son[x];
		dfs2(son[x]);
	}
	for(int i=h[x];i;i=w[i].next)
	{
		int to=w[i].to;
		if(top[to]) continue;
		top[to]=to;
		seg[to]=++seg[0];
		rev[seg[0]]=to;
		dfs2(to);
	}
}
void add2(int x,int k)
{
//	cout<<x<<endl;
	while(x<=n)
	{
		tree[x]+=k;
		x+=(x&-x);
	}
}
int sum(int x)
{
	int ans=0;
	while(x)
	{
		ans+=tree[x];
		x-=(x&-x);
	}
	return ans;
}
void uprange(int x,int y,int z)
{
//	cout<<x<<" "<<y<<" "<<z<<endl;
	while(top[x]!=top[y])
	{
		if(deep[top[x]]<deep[top[y]]) swap(x,y);
		add2(seg[top[x]],z);
		add2(seg[x]+1,-z);
		x=prt[top[x]];
	}
	if(deep[x]>deep[y]) swap(x,y);
	add2(seg[x],z);
	add2(seg[y]+1,-z);
}
int main()
{
	while(~scanf("%d%d%d",&n,&m,&q))
	{
		memset(tree,0,sizeof(tree));
		for(int i=1;i<=n;i++) scanf("%d",&a[i]);
		for(int i=1;i<=m;i++)
		{
			int x,y;
			scanf("%d%d",&x,&y);
			add(x,y);
			add(y,x);
		}
		dfs(1,0);
		seg[0]=seg[1]=rev[1]=top[1]=1;
		dfs2(1);
		for(int i=1;i<=n;i++) add2(seg[i],a[i]),add2(seg[i]+1,-a[i]);
		for(int i=1;i<=q;i++)
		{
			char s[10];
			scanf("%s",s);
			if(s[0]=='Q')
			{
				int x;
				scanf("%d",&x);
				printf("%d\n",sum(seg[x]));
			}
			else if(s[0]=='D')
			{
				int l,r,z;
				scanf("%d%d%d",&l,&r,&z);
				uprange(l,r,-z);
			}
			else
			{
				int l,r,z;
				scanf("%d%d%d",&l,&r,&z);
				uprange(l,r,z);
			}
		}
	}
}