hdu 3078(LCA+排序)
生活随笔
收集整理的這篇文章主要介紹了
hdu 3078(LCA+排序)
小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
題意:一棵無根樹,輸入點(diǎn)數(shù)和操作數(shù),下面一行n個(gè)值代表每個(gè)點(diǎn)的權(quán)。下面n-1行是樹邊
操作分為
0 x w ,表示把點(diǎn)x的權(quán)改為w
k a b , 求出,從a到b的路徑中,第k大的點(diǎn)權(quán)
解題思路:這道題沒什么特別難的地方,只需要找到兩點(diǎn)的LCA,然后回溯上去把路徑上所有的點(diǎn)都拿出來排序即可。在回溯的過程中,需要有一個(gè)記錄該節(jié)點(diǎn)父親節(jié)點(diǎn)的fa[]數(shù)組,在dfs遍歷的時(shí)候可以直接把fa[]數(shù)組更新。。。
這題最開始想會(huì)不會(huì)是樹鏈剖分,看來想多了。
這題用C++可以A,G++超時(shí)。。。
#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; }
總結(jié)
以上是生活随笔為你收集整理的hdu 3078(LCA+排序)的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 【插件发布】JAVA微服务框架,Jeec
- 下一篇: hdu 1078(记忆化搜索)