bzoj 1602: [Usaco2008 Oct]牧场行走（暴力LCA）

1602: [Usaco2008 Oct]牧场行走

Time Limit: 5 Sec   Memory Limit: 64 MB
Submit: 2011   Solved: 1058
[ Submit][ Status][ Discuss]

Description

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)对奶牛之间的距离。

Input

*第一行：两个被空格隔开的整数：N和Q

 *第二行到第n行：第i+1行有两个被空格隔开的整数：AI，BI，LI

*第n+1行到n+Q行：每一行有两个空格隔开的整数：P1，P2，表示两头奶牛的编号。

Output

*第1行到第Q行：每行输出一个数，表示那两头奶牛之间的距离。

Sample Input

4 2

2 1 2

4 3 2

1 4 3

1 2

3 2

Sample Output

2

7

LCA模板题

用在线的，不用倍增，最好写的方法了

#include<stdio.h>
#include<vector>
#include<algorithm>
using namespace std;
vector<int> G[1005];
int road[1005][1005], dep[1005], len[1005], fa[1005];
void Sech(int u, int p)
{
	int v, i;
	fa[u] = p;
	dep[u] = dep[p]+1;
	len[u] = road[u][p];
	for(i=0;i<G[u].size();i++)
	{
		v = G[u][i];
		if(v==p)
			continue;
		Sech(v, u);
	}
}
int LCA(int u, int v)
{
	int ans;
	ans = 0;
	if(dep[u]<dep[v])
		swap(u, v);
	while(dep[u]!=dep[v])
	{
		ans += len[u];
		u = fa[u];
	}
	while(u!=v)
	{
		ans += len[u]+len[v];
		u = fa[u], v = fa[v];
	}
	return ans;
}
int main(void)
{
	int n, q, i, u, v, val;
	while(scanf("%d%d", &n, &q)!=EOF)
	{
		for(i=1;i<=n;i++)
			G[i].clear();
		for(i=1;i<=n-1;i++)
		{
			scanf("%d%d%d", &u, &v, &val);
			G[u].push_back(v);
			G[v].push_back(u);
			road[u][v] = road[v][u] = val;
		}
		Sech(1, 0);
		while(q--)
		{
			scanf("%d%d", &u, &v);
			printf("%d\n", LCA(u, v));
		}
	}
	return 0;
}