【树链剖分】语文

                                              【题目背景】

那个很神奇的诗歌鉴赏树上每一个知识点都有一个重要度,但是捏,新课标那个不和谐的东西决定改变语文考试的侧重点。 
新课标一共有q次操作,分别为以下两种: 
1. 将以知识点i为根的子树除i之外的点的重要度增加delta 
2. 询问当前点重要度。 

【输入格式】

第一行,知识点数n,操作数q. 
第二行,第一个知识点的重要度(这个知识点是根) 
以后n-1行,每行2个数,分别为该知识点的重要度和该知识点的父亲 
以后m行,为一个字符和若干个整数,若字符为p,紧接着有两个整数,整数i和整数delta,表示将以知识点i为根的子树除i之外的点的重要度增加delta;若字符为u,紧接着为一个整数x,表示询问当前点x的重要度。 

【输出格式】

对于每个询问,输出重要度。


树链剖分的水题,通过size直接实现对子树的操作,因为只需要区间修改和单点查询,所以我们使用较为简单的树状数组

#include<iostream>
#include<iomanip>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<queue>
using namespace std;
int n,m,v[500005],h[500005],cnt,deep[500005],size[500005],top[500005],seg[500005],prt[500005],son[500005],tree[500005];
struct edge
{
	int to,next;
}w[500005];
void add(int x,int y)
{
	cnt++;
	w[cnt].to=y;
	w[cnt].next=h[x];
	h[x]=cnt;
}
void dfs(int x)
{
	size[x]=1;
	for(int i=h[x];i;i=w[i].next)
	{
		int to=w[i].to;
		deep[to]=deep[x]+1;
		prt[to]=x;
		dfs(to);
		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];
		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];
		dfs2(to);
	}
}
void addsum(int x,int k)
{
	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;
}
int main()
{
	scanf("%d%d",&n,&m);
	scanf("%d",&v[1]);
	for(int i=2;i<=n;i++)
	{
		scanf("%d",&v[i]);
		int x;
		scanf("%d",&x);
		add(x,i);
	}
	deep[1]=1;
	dfs(1);
	seg[0]=seg[1]=top[1]=1;
	dfs2(1);
	for(int i=1;i<=n;i++) addsum(seg[i],v[i]),addsum(seg[i]+1,-v[i]);
	for(int i=1;i<=m;i++)
	{
		char s[10];
		scanf("%s",s);
		if(s[0]=='u')
		{
			int x;
			scanf("%d",&x);
			printf("%d\n",sum(seg[x]));
		}
		else
		{
			int x,y;
			scanf("%d%d",&x,&y);
			addsum(seg[x]+1,y);
			addsum(seg[x]+size[x],-y);
		}
	}
}

猜你喜欢

转载自blog.csdn.net/Dy_Dream/article/details/82872306