题目描述

   小L非常喜欢树。最近,他发现了一棵有趣的树。这棵树有n个节点(1到n编号),节点i有一个初始的权值ai。这棵树的根是节点1。
   这棵树有一个特殊的性质:当你给节点i的权值加 val 的时候,节点i的所有儿子的权值都会加 -val。注意当你给节点i的儿子的权值加 -val 时,节点i的这个儿子的所有儿子的权值都会加 -(-val),以此类推。样例说明可以很好地帮助你理解这个性质。
   有2种操作:
   操作(a).“1 x val”表示给节点x的权值加val。
   操作(b).“2 x”输出节点x当前的权值。
   为了帮助小L更好地理解这棵树,你必须处理m个操作。

输入

第一行包含2个整数n和m。
第二行包含n个整数a1,a2,…,an(1≤ai≤1000)。
接下来的n-1行,每行两个整数u和v(1≤u接下来的m行,每行包含2种操作的一种。每个操作都保证1≤x≤n,1≤val≤1000。
输出

对于每个操作(b),输出一个整数,表示节点x当前的权值。
输入样例复制

5 5
1 2 1 1 2
1 2
1 3
2 4
2 5
1 2 3
1 1 2
2 1
2 2
2 4
输出样例复制

3
3
0
说明

【输入输出样例说明】
初始各个节点的权值依次为[1,2,1,1,2]。 第一个操作给节点2的权值增加3,会给节点2的儿子4、5的权值增加-3。此时各个节点的权值变成[1,5,1,-2,-1]。 第二个操作给节点1的权值增加2,会给节点1的儿子2、3的权值增加-2,然后会给节点2的儿子4、5的权值增加-(-2)。各个节点的权值变成[3,3,-1,0,1]。

【数据说明】
对于50%的数据,1≤n≤2000,1≤m≤2000。
对于100%的数据,1≤n≤100000,1≤m≤100000。
.
.
.
.
.
线段树:

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int maxn=200001;
int in[maxn],out[maxn],point[maxn],head[maxn],tot,t,vis[maxn],a[maxn],num[maxn],flag[maxn];
struct Edge
{
	int to,next;
}edge[2*maxn];
 
struct data
{
	int l,r,p;
	int laz0,laz1,num;
}node[maxn*4];
 
void add(int x,int y)
{
	edge[tot].to=y;
	edge[tot].next=head[x];
	head[x]=tot++;
}
 
void dfs(int x,int k)
{
	vis[x]=1;
	in[x]=++t;
	num[t]=a[x];
	point[t]=x; 
	flag[x]=k;
	for (int i=head[x];i!=-1;i=edge[i].next)
	{
		int y=edge[i].to;
		if (vis[y]) continue;
		dfs(y,k^1);
	}
	out[x]=t;	
}
 
void build(int x,int y,int cnt)
{
	node[cnt].l=x;
	node[cnt].r=y;
	node[cnt].laz0=node[cnt].laz1=node[cnt].p=0;
	if (x==y)
	{
		node[cnt].p=point[x];
		node[cnt].num=num[x];
		return;
	}
	int mid=(x+y)/2;
	build(x,mid,cnt*2);
	build(mid+1,y,cnt*2+1);
}
 

void down(int cnt)
{
	if (node[cnt].laz0==0&&node[cnt].laz1==0) return;
	int tg0=node[cnt].laz0,tg1=node[cnt].laz1;
	node[2*cnt].laz0+=tg0;
	node[cnt*2].laz1+=tg1;
	node[cnt*2+1].laz0+=tg0;
	node[cnt*2+1].laz1+=tg1;
	if (node[cnt*2].p!=0)
	{
		if (flag[node[cnt*2].p]==0) node[cnt*2].num+=tg0-tg1; else node[cnt*2].num+=tg1-tg0;
	}
	if (node[cnt*2+1].p!=0)
	{
		if(flag[node[cnt*2+1].p]==0) node[cnt*2+1].num+=tg0-tg1; else node[cnt*2+1].num+=tg1-tg0;
	}
	node[cnt].laz0=node[cnt].laz1=0;
}
  
void up(int x,int y,int cnt,int val,int k)
{
	if (x==node[cnt].l&&y==node[cnt].r)
	{
		if (k==0)
		{
			node[cnt].laz0+=val;
			if (node[cnt].p!=0)
			{
				if (flag[node[cnt].p]==0) node[cnt].num+=val; else node[cnt].num-=val;
			}	
		} else
		{
			node[cnt].laz1+=val;
			if (node[cnt].p!=0)
			{
				if (flag[node[cnt].p]==1) node[cnt].num+=val; else node[cnt].num-=val;
			}
		}
		return;
	}
	down(cnt);
	int mid=(node[cnt].l+node[cnt].r)/2;
	if (y<=mid) up(x,y,cnt*2,val,k); else 
	if (x>=mid+1) up(x,y,cnt*2+1,val,k); else
	{
		up(x,mid,cnt*2,val,k);
		up(mid+1,y,cnt*2+1,val,k);
	}
}
 
int find(int x,int y,int cnt)
{	
	if (x==node[cnt].l&&y==node[cnt].r) return node[cnt].num;
	down(cnt);
	int mid=(node[cnt].l+node[cnt].r)/2;
	if (y<=mid) return find(x,y,cnt*2); else 
	if (x>=mid+1) return find(x,y,cnt*2+1);
}
 
int main()
{
	int n,m;
	scanf("%d%d",&n,&m);
	memset(head,-1,sizeof head);
	memset(vis,0,sizeof vis);
	tot=0;
	t=0;
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	for(int i=1;i<n;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		add(x,y);
		add(y,x);
	}
	dfs(1,0);
	build(1,n,1);
	for (int i=1;i<=m;i++)
	{
		int op,x,val;
		scanf("%d%d",&op,&x);
		if(op==1)
		{
			scanf("%d",&val);
			up(in[x],out[x],1,val,flag[x]);
		} else printf("%d
",find(in[x],in[x],1));
	}
	return 0;
}

.
.
.
.
.
树状数组:

#include<iostream>
#include<cstdio>
using namespace std;
int n,m,t,tot,l[100001],r[100001],head[100001],flag[100001],f[100001],a[100001];

struct Edge
{
	int to,next;
}edge[2*200001];

void add(int x,int y)
{
	tot++;
	edge[tot].to=y;
	edge[tot].next=head[x];
	head[x]=tot;
	
	tot++;
	edge[tot].to=x;
	edge[tot].next=head[y];
	head[y]=tot;
}

void dfs(int x,int k)
{
	t++;
	l[x]=t;
	flag[x]=-flag[k];
	for (int i=head[x];i;i=edge[i].next)
	{
		int y=edge[i].to;
		if (y!=k) dfs(y,x);
	}
	r[x]=t;
}

void ins(int x,int y)
{
	for (int i=x;i;i-=i & -i) 
		f[i]+=y;
}
int query(int x)
{
	int sum=0;
	for (int i=x;i<=n;i+=i & -i) 
		sum+=f[i];
	return sum;
}

int main()
{
	scanf("%d%d",&n,&m);
	for (int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	for (int i=1;i<=n-1;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		add(x,y);
	}
	flag[0]=-1;
	dfs(1,0);
	
	for (int i=1;i<=m;i++)
	{
		int rp,x;
		scanf("%d%d",&rp,&x);
		if (rp==1)
		{
			int y;
			scanf("%d",&y);
			ins(r[x],y*flag[x]);
			ins(l[x]-1,-y*flag[x]);
		} else printf("%d
",a[x]+query(l[x])*flag[x]);
	}
	return 0;
}
原文地址:https://www.cnblogs.com/YYC-0304/p/10292843.html