DFS序--树的问题转化为区间问题

问题概述:一颗树有n个结点,编号为1到n,其中1为根节点,现有两种操作:1 x y表示将x结点的权值加上y,2 x表

示查询x到根节点中所有结点的权值和,每个结点权值初始都为0,n和m都小于50000

(http://acm.zzuli.edu.cn/zzuliacm/problem.php?cid=1159&pid=2)

输入样例:                                  对应输出:

1                                                  0

5 5                                               3

1 2                                               2

1 3

3 4

3 5

2 5

1 1 2

1 3 1

2 4

2 1


解题步骤:

①对该树从根节点开始搜索,初始化step==0,每当循环到一个点k时,记录s[k].down==++step,当循环完该点及其

所有子节点准备回溯时,记录s[k].up==++step,那么树对应的区间a[s[k].down]存的便是k点权值的负数,a[s[k].up]

存的便是k点的权值

②对于树的单节点修改--即修改区间中a[s[k].down]和a[s[k].up]的值,

对于树的查询--即查询a[s[k].up]到a[2*n]中间所有值之和

③对于②操作用线段树或树状数组即可


例如对于n==5的树:1->2,1->3,3->4,3->5,第n个点的权值为n,

转化为区间就是:-1 -2 2 -3 -4 4 -5 5 3 1


#include
#include
#include
using namespace std;
typedef struct
{
	int up;
	int down;
}Loct;
Loct s[50005];
int n, step;
long long in[100005];
vector v[50005];
void Sech(int x, int p);
int lowbit(int k)
{
	return k&-k;
}
void Update(int k, int y)
{
	while(k<=2*n)
	{
		in[k] += y;
		k += lowbit(k);
	}
}
long long Query(int k)
{
	long long sum;
	sum = 0;
	while(k>=1)
	{
		sum += in[k];
		k -= lowbit(k);
	}
	return sum;
}
int main(void)
{
	int T, m, i, a, b, t;
	scanf("%d", &T);
	while(T--)
	{
		scanf("%d%d", &n, &m);
		step = 0;
		memset(in, 0, sizeof(in));
		for(i=1;i<=n;i++)
			v[i].clear();
		for(i=1;i<=n-1;i++)
		{
			scanf("%d%d", &a, &b);
			v[a].push_back(b), v[b].push_back(a);
		}
		Sech(1, 0);
		for(i=1;i<=m;i++)
		{
			scanf("%d", &t);
			if(t==1)
			{
				scanf("%d%d", &a, &b);
				Update(s[a].down, -b);
				Update(s[a].up, b);
			}
			if(t==2)
			{
				scanf("%d", &a);
				printf("%lld\n", Query(n*2)-Query(s[a].up-1));
			}
		}
	}
	return 0;
}

void Sech(int x, int p)
{
	int i, temp;
	s[x].down = ++step;
	for(i=0;i


你可能感兴趣的:(技巧)