题意:有一棵以1为根节点共n个节点的的树,每一个节点上都一个value值,执行m次操作。
现在有2种操作:
1.查询某一个节点的当前值
2.给某一个节点加上一个val值,而每一个节点加上val,其儿子节点就需要-val
1 ≤ n, m ≤ 200000
策略:考虑到如果树是链形的 给根节点+val后,每一个点都要执行一次更新。最差会达到10^10级的复杂度。
所以在此基础上考虑成段维护。在每一次跟新操作,需要更新的是当前节点及其子树的值,所以我们可以用dfs序将树序转化为数组序,就可以知道每个节点对应的子树对应在数组中的范围。在更新时用线段树维护相应数组区域val值变化。
然后观察到,如果在奇数层+val,那么之后所有的奇数层+val,偶数层-val。偶数层+val同理。所以用2个线段树去维护奇数层节点和偶数层节点,即在维护奇数层的线段树中,偶数层节点的val值是没有意义的。每次在奇数层+val之后,我们就会去偶数层的线段树-val,反之亦然。然后每次查询奇数层节点去相应的线段树查找,偶数亦然。
但是如果更加仔细的观察发现,即我们如若要对奇数层+val 我们实际+(-val)值,对偶数层+val我们仍然+val这样来操作。然后在我们取奇数节点的时候 我们直接减去维护的值,即-val,而偶数层仍然+val,结果仍然正确。那么用一棵线段树我们就能做到维护了。
查询时将每层的val值自上而下求和后自身相加。
每次操作的时间复杂度都在logn
所以时间复杂度在mlog(n)即可满足要求
发现代码中lazytag可以并不用直接用sg_tree即可。浪费了一些空间。
<pre name="code" class="cpp">//Auhor:Wenjun Shi
//On 2016/7/14
#include <iostream>
#include <cstdio>
#include <vector>
#define MAX_N 200005
using namespace std;
int sg_tree[MAX_N*2],lazytag[MAX_N*2];//懒惰标记
bool niv[MAX_N];//深度奇偶
int in[MAX_N], out[MAX_N];//入出时间戳
int t;
int point[MAX_N];//dfs序中第i个数在线段树中的序号
int n,m,A[MAX_N];
vector<int> G[MAX_N];
void dfs(int u,int fa)
{
// sq[++pos]=u;
in[u]=++t;
for(int i=0; i<G[u].size(); i++)
{
int v=G[u][i];
if(v!=fa)
{
niv[v]=!niv[u];
dfs(v,u);
}
}
out[u]=t;
}
void add_edge(int u,int v)
{
G[u].push_back(v);
G[v].push_back(u);
}
void build_tree(int l,int r,int i)
{
if(l==r) {point[l]=i;return;}
int mid=(l+r)/2;
build_tree(l,mid,i+i);
build_tree(mid+1,r,i+i+1);
}
void add(int lt,int rt,int l,int r,int i,int val)
{
if(rt<l||lt>r) return;
if(lt<=l&&rt>=r)
{
lazytag[i]+=val;
return ;
}
int mid=(l+r)/2;
add(lt,rt,l,mid,i+i,val);
add(lt,rt,mid+1,r,i+i+1,val);
}
int query(int q,int flag)
{
int sum=0;
while(q!=0)
{
sum+=flag*lazytag[q];
q/=2;
}
return sum;
}
int main()
{
int u,v,op,node,val;
scanf("%d%d",&n,&m);
for(int i=1; i<=n; i++) scanf("%d",&A[i]);
for(int i=0; i<n-1; i++)
{
scanf("%d%d",&u,&v);
add_edge(u,v);
}
t=0;
dfs(1,0);
build_tree(1,n,1);
for(int i=0; i<m; i++)
{
scanf("%d%d",&op,&node);
if(op==1)
{
scanf("%d",&val);
add(in[node],out[node],1,n,1,(niv[node]&1)?-val:val);
}
if(op==2)
{
printf("%d\n",A[node]+query(point[in[node]],(niv[node]&1)?-1:1));
}
}
return 0;
}