hdu 3078(LCA+排序)

来源：互联网发布：java web完整项目源码编辑：程序博客网时间：2024/05/29 14:08

题意：一棵无根树，输入点数和操作数，下面一行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;}

0 0