HDU - 3078 Network 倍增LCA
生活随笔
收集整理的這篇文章主要介紹了
HDU - 3078 Network 倍增LCA
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
倍增lca,把路徑上點全部拉出來拍個序輸出第k大就過了。
不知道怎么過的,咱也不敢問。
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<set>
#include<stack>
#include<vector>
#include<map>
#include<queue>
#define myself i,l,r
#define lson i<<1
#define rson i<<1|1
#define Lson i<<1,l,mid
#define Rson i<<1|1,mid+1,r
#define half (l+r)/2
#define inff 0x3f3f3f3f
#define lowbit(x) x&(-x)
#define PI 3.14159265358979323846
#define me(a,b) memset(a,b,sizeof(a))
#define min4(a,b,c,d) min(min(a,b),min(c,d))
#define min3(x,y,z) min(min(x,y),min(y,z))
#define pii make_pair
const int dir[4][2]= {0,-1,-1,0,0,1,1,0};
typedef long long ll;
const ll inFF=9223372036854775807;
typedef unsigned long long ull;
using namespace std;
const int maxn=1e5+5;
int f[maxn][30],head[maxn],val[maxn],ans[maxn],d[maxn];
int sign,n,q,flag=0;struct node
{int to,p;
}edge[maxn<<1];
void add(int u,int v)
{edge[sign]=node{v,head[u]};head[u]=sign++;
}
void init()
{sign=0;for(int i=0;i<=n;i++)head[i]=-1;
}
bool cmp(int x,int y)
{return x>y;
}
void dfs(int u)
{for(int i=1;(1<<i)<=n;i++)f[u][i]=f[f[u][i-1]][i-1];for(int i=head[u];~i;i=edge[i].p){int v=edge[i].to;if(v==f[u][0]) continue;d[v]=d[u]+1;f[v][0]=u;dfs(v);}
}
int get_lca(int a,int b)
{if(d[a]<d[b]) swap(a,b);int x=d[a]-d[b];for(int i=0;(1<<i)<=x;i++)if((1<<i)&x) a=f[a][i];if(a!=b){for(int i=(int)log2(n);i>=0;i--)if(f[a][i]!=f[b][i]) a=f[a][i],b=f[b][i];a=f[a][0];}return a;
}
int main()
{int x,y,z;while(cin>>n>>q){if(flag++) printf("\n");init();for(int i=1;i<=n;i++) scanf("%d",&val[i]);for(int i=1;i<n;i++) scanf("%d %d",&x,&y),add(x,y),add(y,x);f[1][0]=1,d[1]=1,dfs(1);while(q--){scanf("%d %d %d",&z,&x,&y);if(z==0) val[x]=y;else{int k=0;int lca=get_lca(x,y);while(x!=lca) ans[++k]=val[x],x=f[x][0];while(y!=lca) ans[++k]=val[y],y=f[y][0];ans[++k]=val[lca];if(z>k) printf("invalid request!\n");else {sort(ans+1,ans+1+k,cmp);printf("%d\n",ans[z]);}}}}return 0;
}
?
總結
以上是生活随笔為你收集整理的HDU - 3078 Network 倍增LCA的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: POJ - 1986 Distance
- 下一篇: POJ - 2763 Housewife