hdu 3078(LCA+排序)

题意：一棵无根树，输入点数和操作数，下面一行n个值代表每个点的权。下面n-1行是树边

操作分为

0 x w ，表示把点x的权改为w

k a b ， 求出，从a到b的路径中，第k大的点权

解题思路：这道题没什么特别难的地方，只需要找到两点的LCA，然后回溯上去把路径上所有的点都拿出来排序即可。在回溯的过程中，需要有一个记录该节点父亲节点的fa[]数组，在dfs遍历的时候可以直接把fa[]数组更新。。。

这题最开始想会不会是树链剖分，看来想多了。

这题用C++可以A，G++超时。。。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;

const int maxn = 80005;
struct Edge
{
	int to,next;
}edge[maxn<<1];
int n,q,cnt,num,tot,pre[maxn],router[maxn],fa[maxn];
int dp[maxn<<1][20],ver[maxn<<1],R[maxn<<1],first[maxn],tmp[maxn];

void addedge(int u,int v)
{
	edge[num].to = v;
	edge[num].next = pre[u];
	pre[u] = num++;
}

void dfs(int u,int dep)
{
	ver[++cnt] = u; first[u] = cnt; R[cnt] = dep;
	for(int i = pre[u]; i != -1; i = edge[i].next)
	{
		int v = edge[i].to;
		if(fa[u] == v) continue;
		fa[v] = u;
		dfs(v,dep+1);
		ver[++cnt] = u; R[cnt] = dep;
	}
}

int _min(int l,int r)
{
	if(R[l] < R[r]) return l;
	return r;
}

void initRMQ()
{
	for(int i = 1; i <= cnt; i++)
		dp[i][0] = i;
	for(int j = 1; (1 << j) <= cnt; j++)
		for(int i = 1; i + (1 << j) < cnt; i++)
		{
			dp[i][j] = _min(dp[i][j-1],dp[i+(1<<(j-1))][j-1]);
		}
}

int findLCA(int l,int r)
{
	int k = (int)(log(r - l + 1.0) / log(2.0));
	return ver[_min(dp[l][k],dp[r-(1<<k)+1][k])];
}

bool cmp(int a,int b)
{
	return a > b;
}

int main()
{
	int k,a,b;
	while(scanf("%d%d",&n,&q)!=EOF)
	{
		cnt = num = 0;
		memset(pre,-1,sizeof(pre));
		for(int i = 1; i <= n; i++)
			scanf("%d",&router[i]);
		for(int i = 1; i < n; i++)
		{
			scanf("%d%d",&a,&b);
			addedge(a,b);
			addedge(b,a);
		}
		dfs(1,0);
		initRMQ();
		while(q--)
		{
			scanf("%d%d%d",&k,&a,&b);
			if(k == 0)
				router[a] = b;
			else
			{
				int x = first[a], y = first[b];
				if(x > y) swap(x,y);
				int lca = findLCA(x,y);
				tot = 0;
				while(a != lca)
				{
					tmp[++tot] = router[a];
					a = fa[a];
				}
				while(b != lca)
				{
					tmp[++tot] = router[b];
					b = fa[b];
				}
				tmp[++tot] = router[lca];
				if(k > tot) printf("invalid request!\n");
				else
				{
					sort(tmp+1,tmp+1+tot,cmp);
					printf("%d\n",tmp[k]);
				}
			}
		}
	}
	return 0;
}