http://acm.hdu.edu.cn/showproblem.php?pid=2586
问题概述:有一棵含有n个点的树,每个边都有一个权值,之后对于每次查询(a,b)求出从a到b边的长度
输入样例: 对应输出:
2 10
3 2 25
1 2 10
3 1 15
1 2
2 3
离线求解:将所有的查询记录下来,之后通过遍历整棵树来求出所有查询的答案
主要算法:BFS+并查集
flag[k]:k点是否已经遍历; len[k]:该点与根的距离; ans[i]:用来存第i次查询的答案,其中a点与b点的距离
ans[i]==len[a]+len[b]-2*len[a与b的公共祖先]
解题过程:
①:从根开始搜索,搜索到每个点标记flag[k]=1,遍历它所有的子节点
②:当遍历完k的其中一个子节点之后,将这个点与k加入并查集
③:当遍历完k所有的子节点之后,判定有没有端点为k的边要查询,若有,则判定这条边另外一个端点temp是否已
经被遍历过,符合条件的话这个边的长度就可以用上面的公式求出,其中k与temp的公共祖先就是Find(temp)!
#include<stdio.h>
#include<string.h>
#include<vector>
using namespace std;
typedef struct
{
int y;
int val;
}Line;
Line now;
int n, ans[205], ufs[44444], flag[44444], len[44444];
vector<Line> v[44444], q[44444];
void Trajan(int x, int val);
int Find(int x)
{
if(ufs[x]==-1)
return x;
return ufs[x] = Find(ufs[x]);
}
int main(void)
{
int T, i, m, a, b, c;
scanf("%d", &T);
while(T--)
{
scanf("%d%d", &n, &m);
for(i=1;i<=n;i++)
v[i].clear(), q[i].clear();
memset(ufs, -1, sizeof(ufs));
memset(flag, 0, sizeof(flag));
memset(len, 0, sizeof(len));
for(i=1;i<n;i++)
{
scanf("%d%d%d", &a, &b, &c);
now.y = b, now.val = c;
v[a].push_back(now);
now.y = a;
v[b].push_back(now);
}
for(i=1;i<=m;i++)
{
scanf("%d%d", &a, &b);
now.y = b, now.val = i;
q[a].push_back(now);
now.y = a;
q[b].push_back(now);
}
Trajan(1,0);
for(i=1;i<=m;i++)
printf("%d\n", ans[i]);
}
return 0;
}
void Trajan(int x, int val)
{
int i, temp, t1, t2;
flag[x] = 1;
len[x] = val;
for(i=0;i<v[x].size();i++)
{
temp = v[x][i].y;
if(flag[temp]==1)
continue;
Trajan(temp, val+v[x][i].val);
t1 = Find(x); /*注意顺序,否则下面的Find(temp)就会因为路径压缩导致错误*/
t2 = Find(temp);
if(t1!=t2)
ufs[t2] = t1;
}
for(i=0;i<q[x].size();i++)
{
temp = q[x][i].y;
if(flag[temp]==1)
{
//printf("%d %d %d\n", x, temp, Find(temp));
ans[q[x][i].val] = len[x]+len[temp]-2*len[Find(temp)];
}
}
}