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;
}