【9.12模拟赛T4】树【dfs&树状数组】

1140.树

时间限制:10000MS
内存限制:512000KB

题目描述

   小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

分析：

递归 可得$80$分 根据题意 去暴搜即可
正解为线段树或树状数组
但：$1\le n\le 100000，1\le m\le 100000$

80分CODE：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=100005;
int n,m,tot,a[N],x,y,qwq,head[N],u,v;
struct node{
	int to,next;
}edge[N];
void add(int x,int y)
{
	edge[++tot]=(node){y,head[x]};  //邻接表
	head[x]=tot;
}
void ovo(int x,int y)
{
	for(int i=head[x];i;i=edge[i].next)
	{
		int q=edge[i].to;
		a[q]+=y;  //递归搜索
		ovo(q,-y);
	}
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&u,&v);
		add(u,v);
	}
	for(int i=1;i<=m;i++)
	{
		scanf("%d",&qwq);
		if(qwq==1)
		{
			scanf("%d%d",&x,&y);
			a[x]+=y;
			ovo(x,-y);
		}else if(qwq==2)  //题意
		{
			scanf("%d",&x);
			printf("%d\n",a[x]);
		}
	}
	return 0;
}

树状数组：

把$n$个点按照深度的奇偶性分类 不难发现同为奇或者同为偶的那些点每次都同加或者同减
如果$i$的深度为奇数 我们令$flag[i]=1$
如果$i$的深度为偶数 我们令$flag[i]=-1$
对于一个修改操作“$1$ $x$ $val$”
我们给以$x$为根的那整棵子树的每个点加上$val\ast flag[x]$ 这可以用$DFS$序$+$树状数组在$O(logn)$的时间复杂度内解决
对于一个询问操作“$2$ $x$”
我们在树状数组中算出$x$的权值$val$ 然后$x$的实际权值为$val\ast flag[x]$

时间复杂度：$O(m\ast logn)$ 空间复杂度：$O(n)$

CODE：

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=100005;
int n,m,tot,head[N],c[N],a[N],l[N],r[N],stat[N];
struct node{
	int to,next;
}edge[N*2];
int lowbit(int x){return x&(-x);}
void add(int x,int y)
{
	edge[++tot]=(node){y,head[x]};  //邻接表
	head[x]=tot;
}
void dfs(int x,int fa)
{
	stat[x]=-stat[fa];
	l[x]=++tot;
	for(int i=head[x];i;i=edge[i].next)
	{
		if(edge[i].to==fa) continue;
		dfs(edge[i].to,x);
	}
	r[x]=tot;  //求DFS序
}
void build(int x,int y)  //树状数组插入
{
	for(;x;x-=lowbit(x))
		c[x]+=y; 
}
int ask(int x)  //查询
{
	int ans=0;
	for(;x<=n;x+=lowbit(x))
		ans+=c[x];
	return ans;
}
int main(){
	scanf("%d%d",&n,&m);
	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);
	}
	stat[0]=-1;  //初始根节点
	tot=0;
	dfs(1,0);
	for(int i=1;i<=m;i++)
	{
		int T;
		scanf("%d",&T);
		if(T==1)
		{
			int x,y;
			scanf("%d%d",&x,&y);
			build(r[x],y*stat[x]);
			build(l[x]-1,-y*stat[x]);  //DFS序+树状数组
		}else if(T==2)
		{
			int x;
			scanf("%d",&x);
			printf("%d\n",a[x]+ask(l[x])*stat[x]);  //查询实际答案
		}
	}
	return 0;
}

总结：

比赛时不想搞$DP$ 就打了$T4$暴搜和$T1$
结果$T1$出了点问题$90$分
更正后 把$T2$的小小$DP$也做了
$T3$有点$ex\dots \dots$ $T4$听奆佬讲解改
$T4$感谢$lyf$奆佬指正 喜提$100$分

结果：

更正：