首页 > 代码库 > BZOJ 3364: [Usaco2004 Feb]Distance Queries 距离咨询
BZOJ 3364: [Usaco2004 Feb]Distance Queries 距离咨询
Description
一棵树,询问两点间距离.
Sol
倍增.
方向没用.
没有然后了.
Code
/************************************************************** Problem: 3364 User: BeiYu Language: C++ Result: Accepted Time:400 ms Memory:11556 kb****************************************************************/ #include<cstdio>#include<utility>#include<vector>#include<iostream>using namespace std; #define mpr make_pairtypedef pair< int,int > pr;const int N = 40005;const int M = 20; int n,m,k;vector<pr> h[N];bool b[N];int pow2[M],d[N];int f[N][M],g[N][M]; inline int in(int x=0,char ch=getchar()){ while(ch>‘9‘ || ch<‘0‘) ch=getchar(); while(ch>=‘0‘ && ch<=‘9‘) x=(x<<3)+(x<<1)+ch-‘0‘,ch=getchar();return x; }void DFS(int u,int fa){ b[u]=1,d[u]=d[fa]+1; for(int i=0,v;i<h[u].size();i++) if((v=h[u][i].first)!=fa){ DFS(v,u),f[v][0]=u,g[v][0]=h[u][i].second; }}void work(){ pow2[0]=1;for(int i=1;i<M;i++) pow2[i]=pow2[i-1]<<1; for(int j=1;j<M;j++) for(int i=1;i<=n;i++) f[i][j]=f[f[i][j-1]][j-1],g[i][j]=g[i][j-1]+g[f[i][j-1]][j-1];}int Dis(int u,int v){ if(d[u]<d[v]) swap(u,v); int l=d[u]-d[v],res=0; for(int i=0;i<M;i++) if(l&pow2[i]) res+=g[u][i],u=f[u][i]; if(u==v) return res; for(int i=M-1;~i;--i){ if(f[u][i]!=f[v][i]) res+=g[u][i]+g[v][i],u=f[u][i],v=f[v][i]; }return g[u][0]+g[v][0]+res; }int main(){ n=in(),m=in(); for(int i=1,u,v,w;i<=m;i++) u=in(),v=in(),w=in(),h[u].push_back(mpr(v,w)),h[v].push_back(mpr(u,w)); for(int i=1;i<=n;i++) if(!b[i]) DFS(i,i); work(); for(k=in();k--;){ int u=in(),v=in(); printf("%d\n",Dis(u,v)); }return 0;}
BZOJ 3364: [Usaco2004 Feb]Distance Queries 距离咨询
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。