bzoj 3364: [Usaco2004 Feb]Distance Queries 距离咨询 LCA

→题目链接←

【想说的话】

TLE了3遍后我深刻的认识到了....

跑最短路果然不可行...

【题解】

显然是裸的LCA

【代码】

#include<iostream>
#include<cstdio>
#include<queue>
#include<vector>

using namespace std;

struct node{
int to,len;
node(int x,int y){to=x,len=y;}
};

int n,m;
vector<node>v[40040];
int fa[40040];
int deep[40040];
int par[40040][20];
int dis[40040];

void dfs(int x){
for(int i=0; i<v[x].size(); i++){
int to=v[x][i].to;
int len=v[x][i].len;
if(fa[x]==to)continue;
fa[to]=x;
deep[to]=deep[x]+1;
par[to][0]=x;
dis[to]=dis[x]+len;
dfs(to);
}
}

void cal(){
for(int j=1; (1<<j)<=n; j++){
for(int i=1; i<=n; i++){
par[i][j]=par[par[i][j-1]][j-1];
}
}
}

int lca(int x,int y){
if(deep[x]>deep[y])swap(x,y);
int num=deep[y]-deep[x];
for(int i=18; i>=0; i--){
if(num>=(1<<i)){
y=par[y][i];
num-=(1<<i);
}
}
if(x!=y){
for(int i=18; i>=0; i--){
if(par[x][i]!=par[y][i]){
x=par[x][i];
y=par[y][i];
}
}
x=par[x][0];
}
return x;
}

int main(){
scanf("%d%d",&n,&m);
for(int i=0; i<m; i++){
int x,y,z;
char c[2];
scanf("%d%d%d%s",&x,&y,&z,c);
v[x].push_back(node(y,z));
v[y].push_back(node(x,z));
}
dfs(1);
cal();
int x;
scanf("%d",&x);
for(int i=0; i<x; i++){
int y,z,xx;
scanf("%d%d",&y,&z);
xx=lca(y,z);
printf("%d\n",dis[y]+dis[z]-2*dis[xx]);
}

return 0;
}