hdu 3966 (树链剖分+线段树)

题意:给出一棵树,每个节点有一些敌人,有三种操作,I:x,y,路径上的所有点的人数+w。D:x,y,路径上的所有点的人数-w。Q:节点x的人数。

思路:如果是线性的线段树可以做,所以要树链剖分,剖分后的每条链是线性的,更改区间时用找公共祖先的方式,将两点找到同一条链上,同一条链上的两点间可以直接线段树更改。




#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<stdio.h>
#include<string.h>
#include <iostream>
using namespace std;
const int N=51000;
int son[N],sz[N],dep[N],top[N],father[N],head[N],ti[N],num,idx,a[N];
struct edge
{
	int ed,next;
}e[N*4];
void addedge(int x,int y)
{
	e[num].ed=y;e[num].next=head[x];head[x]=num++;
	e[num].ed=x;e[num].next=head[y];head[y]=num++;
}
//*******************树链剖分**********************
void find_son(int u)
{
	int i,v;
	sz[u]=1;son[u]=0;
	for(i=head[u];i!=-1;i=e[i].next)
	{
		v=e[i].ed;
		if(v==father[u])continue;
		father[v]=u;dep[v]=dep[u]+1;
		find_son(v);
		sz[u]+=sz[v];
		if(sz[v]>sz[son[u]])son[u]=v;
	}
}
void find_time(int u,int fa)
{
	int i,v;
	ti[u]=idx++;	
	top[u]=fa;
	if(son[u]!=0)find_time(son[u],top[u]);
	for(i=head[u];i!=-1;i=e[i].next)
	{
		v=e[i].ed;
		if(v==son[u]||v==father[u])continue;
		find_time(v,v);
	}
}
//************************线段树**************************
struct Tree
{
	int R,L,ct;//ct:该区间的增量
}T[N*4];
void buildTree(int L,int R,int id)
{
	T[id].L=L;T[id].R=R;T[id].ct=0;
	if(L==R) return ;		
	int mid=(L+R)>>1,li=id<<1,ri=li+1;
	buildTree(L,mid,li);
	buildTree(mid+1,R,ri);
}
void insert(int L,int R,int id,int w)
{
	if(T[id].L==L&&T[id].R==R)
	{
		T[id].ct+=w;
		return ;
	}
	int mid=(T[id].L+T[id].R)>>1,li=id<<1,ri=li+1;
	if(mid>=R)insert(L,R,li,w);
	else if(mid<L)insert(L,R,ri,w);
	else 
	{
		insert(L,mid,li,w);
		insert(mid+1,R,ri,w);
	}
}
int find(int i,int id,int w)
{
	if(T[id].L==T[id].R)
	    return w+T[id].ct;
	int mid=(T[id].L+T[id].R)>>1,li=id<<1,ri=li+1;
	if(mid>=i)return find(i,li,w+T[id].ct);
	else return find(i,ri,w+T[id].ct);
}
void lca(int x,int y,int w)
{
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]])swap(x,y);
		insert(ti[top[x]],ti[x],1,w);
		x=father[top[x]];
	}
	if(dep[x]>dep[y])swap(x,y);
	insert(ti[x],ti[y],1,w);
}
int main()
{
	int i,n,m,q,x,y,w;
	char str[10];
	while(scanf("%d%d%d",&n,&m,&q)!=-1)
	{
		memset(head,-1,sizeof(head));
		num=0;
		for(i=1;i<=n;i++)
			scanf("%d",&a[i]);
		for(i=1;i<=m;i++)
		{
			scanf("%d%d",&x,&y);
			addedge(x,y);
		}
		father[1]=sz[0]=0;dep[1]=idx=1;
		find_son(1);
		find_time(1,1);
		buildTree(1,n,1);
		while(q--)
		{
			scanf("%s",str);
			if(str[0]=='Q')
			{
				scanf("%d",&x);
				printf("%d\n",a[x]+find(ti[x],1,0));
			}
			else
			{
				scanf("%d%d%d",&x,&y,&w);
				if(str[0]=='I')	w=-w;				
				lca(x,y,-w);
			}
		}
	}
	return 0;
}

点赞