【LCA】 HDU 3078 Network

题意： 一棵树 点有权值

求u-v 的路径上 第 k 大的点权值（包括u和v）

k==0 为更改操作

先LCA 转 RMQ

如果u==v 判断k==1

否则 求 （u，v）的LCA 然后 求u的前驱 和v的前驱 并且把权值存起来

把权值排序 然后输出

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <string>
#include <iostream>
#include <algorithm>
#include <sstream>
#include <cmath>
using namespace std;
#include <queue>
#include <stack>
#include <vector>
#include <deque>
#include <map>
#define cler(arr, val)    memset(arr, val, sizeof(arr))
typedef long long  LL;
const int MAXN = 100000+6;
const int MAXM = 140000;
const int INF = 0x3f3f3f3f;
const int mod = 1000000007;
int dp[MAXN*2][25];
struct node
{
int v,next;
}edge[MAXN*2];
int len[MAXN];
int head[MAXN],tol,first[MAXN],time,vis[MAXN*2],dep[MAXN*2],pre[MAXN];
void addedge(int u,int v)
{
edge[tol].v=v;
edge[tol].next=head[u];
head[u]=tol++;
}
void RMQ_init(int n)
{
for(int i = 1; i <= n; i++)
{
dp[i][0] = i;
}
for(int j=1;(1<<j)<=n;j++)
{
for(int i=1;i+(1<<j)-1<=n;i++)
{
dp[i][j] = dep[dp[i][j-1]] <
dep[dp[i+(1<<(j-1))][j-1]]?dp[i][j-1]:dp[i+(1<<(j-1))][j-1];
}
}
}
int RMQ(int l,int r)
{
int k=(int)(log(double(r-l+1))/log((double)2));
return dep[dp[l][k]] <= dep[dp[r-(1<<k)+1][k]]?dp[l][k]:dp[r-(1<<k)+1][k];
}
void dfs(int u,int pr,int deep)
{
vis[++time]=u;
dep[time]=deep;
first[u]=time;
for(int i=head[u];~i;i=edge[i].next)
{
int v=edge[i].v;
if(v==pr) continue;
pre[v]=u;
dfs(v,u,deep+1);
vis[++time]=u;
dep[time]=deep;
}
}
void init()
{
cler(head,-1);
tol=0;
time=0;
}
vector<int>g;
int main()
{
#ifndef ONLINE_JUDGE
freopen("in.txt", "r", stdin);
// freopen("out.txt", "w", stdout);
#endif
int n,m,k;
while(cin>>n>>m)
{
init();
int a,b,c;
for(int i=1;i<=n;i++)
scanf("%d",&len[i]);
for(int i=0;i<n-1;i++)
{
scanf("%d%d",&a,&b);
addedge(a,b);
addedge(b,a);
}
dfs(1,0,0);
RMQ_init(time);
for(int i=0;i<m;i++)
{
scanf("%d%d%d",&k,&a,&b);
if(k==0)
len[a]=b;
else
{
if(a==b)
{
if(k==1) printf("%d\n",len[a]);
else puts("invalid request!");
}
else
{
g.clear();
int u=first[a],v=first[b];
if(u>v) swap(u,v);
int c=RMQ(u,v);
c=vis[c];
g.push_back(len[c]);
while(a!=c)
{
g.push_back(len[a]);
a=pre[a];
}
while(b!=c)
{
g.push_back(len[b]);
b=pre[b];
}
if(g.size()<k)
puts("invalid request!");
else
{
sort(g.begin(),g.end());
printf("%d\n",g[g.size()-k]);
}
}
}
}
}
return 0;
}