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