题目链接:http://codeforces.com/gym/101808/problem/K
题面
大意
给出一个有
N
N
N个点以及
N
N
N条边的无向图。
需要处理Q个询问,每个询问会询问从
X
X
X到
Y
Y
Y的最短路径。
思路
既然是最短路问题,那么显然可以用dijkstra或floyd算法来做
考虑到这题
N
N
N最大为105的数据规模,显然不能用总时间复杂度为O(
N
3
N^3
N3)的dijkstra或floyd算法。
注意到这题有
N
N
N个点,
N
N
N条边同时题面并没有说明两点不可互相到达的情况输出什么答案题面说明为连通图,大胆猜测输入数据是保证任意两个点之间可以相互到达的。
那么这个图就可以被看成是一棵有
N
N
N个点,
N
−
1
N-1
N−1条边的树再加上一条额外的边。那么我们只需要拆去这条额外的边就把问题转化成了LCA问题(树上两点间路径唯一确定),可以O(
N
l
o
g
N
NlogN
NlogN)预处理出LCA,查询的时候时间复杂度是O(
l
o
g
N
logN
logN)。
考虑到还有一条被拆掉的边,设这条边的端点是
A
A
A、
B
B
B,长度为
W
W
W,待查询的点为
X
X
X,
Y
Y
Y那么我们还需要计算
M
I
N
(
d
i
s
(
A
,
X
)
+
d
i
s
(
B
,
X
)
,
d
i
s
(
A
,
Y
)
+
d
i
s
(
B
,
Y
)
,
d
i
s
(
A
,
Y
)
+
d
i
s
(
B
,
X
)
+
W
,
d
i
s
(
A
,
X
)
+
d
i
s
(
B
,
Y
)
+
W
)
MIN(dis(A,X)+dis(B,X),dis(A,Y)+dis(B,Y),dis(A,Y)+dis(B,X)+W,dis(A,X)+dis(B,Y)+W)
MIN(dis(A,X)+dis(B,X),dis(A,Y)+dis(B,Y),dis(A,Y)+dis(B,X)+W,dis(A,X)+dis(B,Y)+W)的值并与树上的路径的值进行比较,输出较小的那个值。
AC代码
#include<stdio.h>
#include<algorithm>
#define N 100005
using namespace std;
int t,n,q,x,y,z,a,b,dis;
int head[N],w[N*2],p[N*2],nxt[N*2];
int depth[N],lca[N][20];
long long lca_dis[N][20];
long long value[N];
int find_and_make(int now,int f)
{
int as=0,a2;
lca[now][0]=f;lca_dis[now][0]=value[now];
for(int i=1;i<20;i++) lca[now][i]=lca[lca[now][i-1]][i-1],lca_dis[now][i]=lca_dis[lca[now][i-1]][i-1]+lca_dis[now][i-1];
depth[now]=depth[f]+1;
for(int i=head[now];i;i=nxt[i])
{
if(p[i]!=f)
{
if(depth[p[i]])
{
as=i;
}else
{
value[p[i]]=w[i];
if((a2=find_and_make(p[i],now))!=0)as=a2;
}
}
}
return as;
}
int get_lca(int a,int b)
{
if(depth[a]<depth[b])swap(a,b);
for(int i=19;i+1;i--)
{
if(depth[lca[a][i]]>=depth[b])a=lca[a][i];
}
if(a==b)return a;
for(int i=19;i+1;i--)
{
if(lca[a][i]!=lca[b][i])
{
a=lca[a][i];b=lca[b][i];
}
}
return lca[a][0];
}
long long calc(int a,int b)
{
long long temp=0;
if(depth[a]<depth[b])swap(a,b);
for(int i=19;i+1;i--)
{
if(depth[lca[a][i]]>=depth[b])
{
temp+=lca_dis[a][i];a=lca[a][i];
}
}
if(a==b)return temp;
for(int i=19;i+1;i--)
{
if(lca[a][i]!=lca[b][i])
{
temp+=lca_dis[a][i];
temp+=lca_dis[b][i];
a=lca[a][i];b=lca[b][i];
}
}
return temp+lca_dis[a][0]+lca_dis[b][0];
}
int main()
{
scanf("%d",&t);
while(t--)
{
for(int i=1;i<=n;i++)head[i]=0,depth[i]=0;
scanf("%d%d",&n,&q);
for(int i=1;i<=n;i++)
{
scanf("%d%d%d",&x,&y,&z);
p[i*2-1]=y;w[i*2-1]=z;nxt[i*2-1]=head[x];head[x]=i*2-1;
p[i*2]=x;w[i*2]=z;nxt[i*2]=head[y];head[y]=i*2;
}
int pz=find_and_make(1,0);
x=p[pz];y=p[((pz+1)^1)-1];dis=w[pz];
for (int i=1;i<=q;i++)
{
scanf("%d%d",&a,&b);
long long min=1e17;
int lz=get_lca(a,b);
long long temp,temp2;
if(((temp=calc(a,lz))+(temp2=calc(b,lz)))<min)
min=temp+temp2;
if(((temp=calc(a,x))+(temp2=calc(b,x)))<min)
min=temp+temp2;
if(((temp=calc(a,y))+(temp2=calc(b,y)))<min)
min=temp+temp2;
if(((temp=calc(a,x))+(temp2=calc(b,y))+dis)<min)
min=temp+temp2+dis;
if(((temp=calc(a,y))+(temp2=calc(b,x))+dis)<min)
min=temp+temp2+dis;
printf("%lld\n",min);
}
}
return 0;
}