N头牛(2<=n<=1000)别人被标记为1到n,在同样被标记1到n的n块土地上吃草,第i头牛在第i块牧场吃草。 这n块土地被n-1条边连接。 奶牛可以在边上行走,第i条边连接第Ai,Bi块牧场,第i条边的长度是Li(1<=Li<=10000)。 这些边被安排成任意两头奶牛都可以通过这些边到达的情况,所以说这是一棵树。 这些奶牛是非常喜欢交际的,经常会去互相访问,他们想让你去帮助他们计算Q(1<=q<=1000)对奶牛之间的距离。
N头牛(2<=n<=1000)别人被标记为1到n,在同样被标记1到n的n块土地上吃草,第i头牛在第i块牧场吃草。 这n块土地被n-1条边连接。 奶牛可以在边上行走,第i条边连接第Ai,Bi块牧场,第i条边的长度是Li(1<=Li<=10000)。 这些边被安排成任意两头奶牛都可以通过这些边到达的情况,所以说这是一棵树。 这些奶牛是非常喜欢交际的,经常会去互相访问,他们想让你去帮助他们计算Q(1<=q<=1000)对奶牛之间的距离。
*第一行:两个被空格隔开的整数:N和Q
*第二行到第n行:第i+1行有两个被空格隔开的整数:AI,BI,LI
*第n+1行到n+Q行:每一行有两个空格隔开的整数:P1,P2,表示两头奶牛的编号。
*第1行到第Q行:每行输出一个数,表示那两头奶牛之间的距离。
1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include<cstdlib> 5 #include<cmath> 6 #include<ctime> 7 #include<algorithm> 8 using namespace std; 9 #define MAXN 1010 10 struct node{int y,next,v;}e[MAXN*2]; 11 int n,m,len,Link[MAXN],f[MAXN],deep[MAXN],dis[MAXN],anc[MAXN][25]; 12 inline int read() 13 { 14 int x=0,f=1; char ch=getchar(); 15 while(!isdigit(ch)) {if(ch==‘-‘) f=-1; ch=getchar();} 16 while(isdigit(ch)) {x=x*10+ch-‘0‘; ch=getchar();} 17 return x*f; 18 } 19 void insert(int x,int y,int v) {e[++len].next=Link[x];Link[x]=len;e[len].y=y;e[len].v=v;} 20 void dfs(int x,int fa) 21 { 22 anc[x][0]=f[x]; 23 for(int i=1;i<=20;i++) anc[x][i]=anc[anc[x][i-1]][i-1]; 24 for(int i=Link[x];i;i=e[i].next) 25 if(e[i].y!=fa) 26 { 27 deep[e[i].y]=deep[x]+1; 28 dis[e[i].y]=dis[x]+e[i].v; 29 f[e[i].y]=x; 30 dfs(e[i].y,x); 31 } 32 } 33 int lca(int x,int y) 34 { 35 if(deep[x]<deep[y]) swap(x,y); 36 for(int i=20;i>=0;i--) 37 if(deep[anc[x][i]]>=deep[y]) x=anc[x][i]; 38 if(x==y) return x; 39 for(int i=20;i>=0;i--) 40 if(anc[x][i]!=anc[y][i]) x=anc[x][i],y=anc[y][i]; 41 return f[x]; 42 } 43 int main() 44 { 45 //freopen("cin.in","r",stdin); 46 //freopen("cout.out","w",stdout); 47 n=read(); m=read(); 48 for(int i=1;i<n;i++) {int x=read(),y=read(),v=read(); insert(x,y,v); insert(y,x,v);} 49 deep[1]=1; dfs(1,0); 50 for(int i=1;i<=m;i++) 51 { 52 int x=read(),y=read(); 53 printf("%d\n",dis[x]+dis[y]-2*dis[lca(x,y)]); 54 } 55 return 0; 56 }
原文:http://www.cnblogs.com/chty/p/5971457.html