题目链接：http://www.lydsy.com/JudgeOnline/problem.php?id=3364
题意：k 次询问树上两点距离。
约定
二元组形如 (l,r) 表示树上节点 l 到 r 最短距离。
(x,y)=(x,root)+(y,root)-2*(lca(x,y),root)
于是写了个倍增
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 |
#include <cmath> #include <queue> #include <cstdio> #include <iomanip> #include <cstdlib> #include <cstring> #include <iostream> #include <algorithm> #define N 40010 #define ll long long using namespace std; inline int read() { int x=0,f=1;char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();} return x*f; } struct zgz { int next,to,val; }edge[N<<1]; int head[N],cnt=1; void add(int from,int to,int val) { edge[cnt].to=to; edge[cnt].val=val; edge[cnt].next=head[from]; head[from]=cnt++; } int deep[N],len[N],anc[N][20]; void dfs(int x) { for(int i=head[x];i;i=edge[i].next) { int to=edge[i].to,val=edge[i].val; if(to==anc[x][0])continue ; anc[to][0]=x; deep[to]=deep[x]+1; len[to]=len[x]+val; dfs(to); } } int get_lca(int x,int y) { if(deep[x]>deep[y])swap(x,y); for(int i=16;i>=0;i--) if(deep[anc[y][i]]>=deep[x]) y=anc[y][i]; if(x==y)return x; for(int i=16;i>=0;i--) if(anc[x][i]!=anc[y][i]) x=anc[x][i],y=anc[y][i]; return anc[y][0]; } int n,m; int main() { n=read(),m=read(); for(int i=1;i<n;i++) { int x=read(),y=read(),v=read(); add(x,y,v),add(y,x,v); } deep[1]=1;dfs(1); for(int j=1;j<=16;j++) for(int i=1;i<=n;i++) anc[i][j]=anc[anc[i][j-1]][j-1]; m=read(); while(m--) { int x=read(),y=read(); int z=get_lca(x,y); printf("%d\n",len[x]+len[y]-2*len[z]); } } |
Comments | NOTHING