hdu 3966 Aragorn's Story 樹鏈剖分 bfs實現（不知道爲什麼會RE）

題意：給出一棵樹，增加或減少樹上一條路徑上點的值，詢問點的值

思路：學習了樹鏈剖分，並且使用了紅書的bfs版本，然後映射到線段樹上，手動擴棧後還是一直RE，真的不知道怎麼弄了，紅書上很多數據結構在這道題沒有用上，但也加上了自己的理解

希望各位神犇路過看下，不知道爲什麼會re

題目鏈接：http://acm.hdu.edu.cn/showproblem.php?pid=3966

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int maxn = 50010;

struct edge
{
	int u, v, next;
} e[maxn << 1];
int head[maxn], cnt;
int in[maxn], root;

void init()
{
	cnt = 0;
	memset(in, 0, sizeof(in));
	memset(head, -1, sizeof(head));
}

void addedge(int u, int v)
{
	e[cnt].u = u, e[cnt].v = v, e[cnt].next = head[u], head[u] = cnt++;
}

int siz[maxn], dep[maxn], fa[maxn];
//int len[maxn];//len[x]表示編號爲x的重鏈的長度
int son[maxn];//son[v]表示節點v的重兒子
//int id[maxn];//int id[maxn];//id[x]表示節點x所在的重鏈中的編號，按樹的深度由深至淺編號
//int belong[maxn];//int belong[maxn];//belong[x]表示節點x所在的重鏈編號
//int top[maxn];//int top[maxn];//top[x]表示編號爲x的重鏈的頂端節點編號，那麼一個節點u的重鏈頂端編號就是top[belong[u]]
int tid[maxn];//tid[x]表示x節點及其與父節點的連邊在線段樹的位置
int ran[maxn];//ran[x]表示線段樹的x的位置是哪個節點編號
int apex[maxn];//apex[x]表示x節點的所在重鏈的頂端節點
bool vis[maxn];
int q[maxn];//隊列
int l, r, tot;//tot爲重鏈編號，從1開始編號
int res;//res是在線段樹中的編號

void Clear()
{
	memset(son, 0, sizeof(son));
	memset(dep, -1, sizeof(dep));
	//memset(vis, true, sizeof(vis));
	l = 0, r = 1;
	q[l] = root;
	dep[root] = 0;
	fa[root] = -1;
	tot = 0;
	res = 0;
}

void split(int n)
{
	Clear();

	while (l < r)
	{
		int x = q[l++];
		vis[x] = false;
		for (int i = head[x]; i != -1; i = e[i].next)
		{
			int v = e[i].v;
			if (dep[v] == -1)
			{
				q[r++] = v;
				dep[v] = dep[x] + 1;
				fa[v] = x;
			}
		}
	}
		
	for (int i = n - 1; i >= 0; i--)//對於n個節點的一棵樹來說，跑完bfs後在隊列必然有是有n個點
	{
		//從後往前算，相當於dfs的回溯
		int x = q[i], p = -1;
		siz[x] = 1;
		for (int i = head[x]; i != -1; i = e[i].next)
		{
			int v = e[i].v;
			if (vis[v])
			{
				siz[x] += siz[v];
				if (p == -1 || siz[v] > siz[p])
					p = v;
			}
		}
		if (p == -1)//表明此時的x是葉子節點
		{
			son[x] = -1;
			//tot++;//重鏈編號
			//len[tot] = 1;
			//id[x] = 1;
			//top[tot] = x;
			//belong[x] = tot;
		}
		else//p爲x的重兒子
		{
			son[x] = p;
			//belong[x] = belong[p];//重兒子的重鏈編號自然是父親的重鏈編號
			//len[belong[x]]++;//重鏈長度+1
			//id[x] = len[belong[x]];//x的在這條鏈的編號是目前鏈的長度
			//top[belong[x]] = x;//這條重鏈的頂端節點改爲x
		}
		vis[x] = true;
	}

	//下面這段是將重鏈放置在線段樹中
	memset(vis, false, sizeof(vis));
	for (int i = 0; i < n; i++)
	{
		int x = q[i];
		if (!vis[x])
		{
			apex[x] = x;
			int temp = apex[x];
			while (!vis[x] && x != -1)
			{
				vis[x] = true;
				apex[x] = temp;
				tid[x] = ++res;
				ran[res] = x;
				x = son[x];
			}
		}
	}
}

#define lson l, m, rt << 1
#define rson m + 1, r, rt << 1 | 1

int sum[maxn << 2], num[maxn], add[maxn << 2];
int n, m, p;

void pushdown(int rt, int len)
{
	if (add[rt])
	{
		add[rt << 1] += add[rt];
		add[rt << 1 | 1] += add[rt];
		sum[rt << 1] += add[rt] * (len - (len >> 1));
		sum[rt << 1 | 1] += add[rt] * (len >> 1);
		add[rt] = 0;
	}
}

void pushup(int rt)
{
	sum[rt] = max(sum[rt << 1], sum[rt << 1 | 1]);
}

void build(int l, int r, int rt)
{
	add[rt] = 0;
	if (l == r)
	{
		sum[rt] = num[ran[l]];
		return;
	}
	int m = (l + r) >> 1;
	build(lson);
	build(rson);
	pushup(rt);
}

void update(int L, int R, int val, int l, int r, int rt)
{
	if (L <= l && r <= R)
	{
		add[rt] += val;
		sum[rt] += val * (r - l + 1);
		return;
	}
	pushdown(rt, r - l + 1);
	int m = (l + r) >> 1;
	if (L <= m) update(L, R, val, lson);
	if (m < R) update(L, R, val, rson);
	pushup(rt);
}

void change(int x, int y, int val)
{
	while (apex[x] != apex[y])
	{
		if (dep[apex[x]] < dep[apex[y]])
			swap(x, y);
		update(tid[apex[x]], tid[x], val, 1, n, 1);
		x = fa[apex[x]];
	}
	if (dep[x] > dep[y]) swap(x, y);
	update(tid[x], tid[y], val, 1, n, 1);
}

int query(int l, int r, int rt, int pos)
{
	if (l == r)
		return sum[rt];
	pushdown(rt, r - l + 1);
	int m = (l + r) >> 1;
	int ret = 0;
	if (pos <= m) ret = query(lson, pos);
	else ret = query(rson, pos);
	pushup(rt);
	return ret;
}

int main()
{
	while (~scanf("%d%d%d", &n, &m, &p))
	{
		init();
		for (int i = 1; i <= n; i++)
			scanf("%d", &num[i]);
		for (int i = 0; i < m; i++)
		{
			int u, v;
			scanf("%d%d", &u, &v);
			addedge(u, v);
			in[v]++;
		}
		for (int i = 1; i <= n; i++)//找到樹的根
		{
			if (in[i] == 0)
			{
				root = i;
				break;
			}
		}
		split(n);
		build(1, n, 1);
		char op[10];
		int a, b, c;
		while (p--)
		{
			scanf("%s", op);
			if (op[0] == 'Q')
			{
				scanf("%d", &a);
				printf("%d\n", query(1, n, 1, tid[a]));
			}
			else
			{
				scanf("%d%d%d", &a, &b, &c);
				if (op[0] == 'D') change(a, b, -c);
				else change(a, b, c);
			}
		}
	}
	return 0;
}