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行:每行输出一个数,表示那两头奶牛之间的距离。
资格赛
#include<iostream> #include<cstdio> #include<algorithm> #include<cstring> #define N 100000 using namespace std; int n,m; int next[N],point[N],v[N],c[N],tot,deep[N],mi[20],fa[1003][20]; int len[1003][20]; void add(int x,int y,int z) { tot++; next[tot]=point[x]; point[x]=tot; v[tot]=y; c[tot]=z; tot++; next[tot]=point[y]; point[y]=tot; v[tot]=x; c[tot]=z; } void build(int x,int f,int depth) { deep[x]=depth; for (int i=1;i<=10;i++) { if (deep[x]-mi[i]<=0) break; fa[x][i]=fa[fa[x][i-1]][i-1]; len[x][i]=len[x][i-1]+len[fa[x][i-1]][i-1]; } for (int i=point[x];i;i=next[i]) if (v[i]!=f) { fa[v[i]][0]=x; len[v[i]][0]=c[i]; build(v[i],x,depth+1); } } int lca(int x,int y) { if (deep[x]<deep[y]) swap(x,y); int k=deep[x]-deep[y]; int sum=0; for (int i=0;i<=10;i++) if ((k>>i)&1) sum+=len[x][i],x=fa[x][i]; if (x==y) return sum; for (int i=10;i>=0;i--) if (fa[x][i]!=fa[y][i]) { sum+=len[x][i]; sum+=len[y][i]; x=fa[x][i]; y=fa[y][i]; } sum+=len[x][0]; sum+=len[y][0]; return sum; } int main() { scanf("%d%d",&n,&m); for (int i=1;i<=n-1;i++) { int x,y,z; scanf("%d%d%d",&x,&y,&z); add(x,y,z); } mi[0]=1; for (int i=1;i<=10;i++) mi[i]=mi[i-1]*2; build(1,0,1); for (int i=1;i<=m;i++) { int x,y; scanf("%d%d",&x,&y); printf("%d\n",lca(x,y)); } }