HDU - 2874 Connections between cities
题意:
有n个点,m条带权无向边,q条询问。对每次询问判断两点是否存在LCA,存在输出最短距离,否则输出 Not connected 。
思路:
用并查集判断是否存在LCA
(x,y)距离公式:dis[x] + dis[y] - 2*dis[lca(x,y)]
#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<sstream>
#include<cstring>
#include<bitset>
#include<cstdio>
#include<string>
#include<deque>
#include<stack>
#include<cmath>
#include<queue>
#include<set>
#include<map>
#define mod 1000000007
using namespace std;
typedef long long ll;
const int maxn = 1e4+10;
struct node
{
int to,w;
int next;
}e[2*maxn];
int n,m,q,cnt,N;
int fa[maxn][20]; //表示节点i往上跳2^j次后的节点 注意j的取值范围
int head[maxn];
int d[maxn]; //当前节点所在的深度
int pre[maxn];
int dis[maxn];
void add(int u,int v,int w)
{
e[cnt].to = v;
e[cnt].w = w;
e[cnt].next = head[u];
head[u] = cnt++;
}
void init() //初始化
{
cnt = 0;
N = (int)(log(1.0*n)/log(2))+1;
for(int i=1;i<=n;i++)
pre[i] = i;
memset(head,-1,sizeof(head));
memset(dis,0,sizeof(dis));
memset(fa,0,sizeof(fa));
memset(d,0,sizeof(d));
}
int find(int x) //查找父节点
{
if(x==pre[x])
return x;
else
return pre[x] = find(pre[x]);
}
void merge(int x,int y) //合并
{
int xx = find(x);
int yy = find(y);
if(xx!=yy)
pre[yy] = xx;
}
void dfs(int u,int f)
{
for(int i=1;i<=N;i++)
fa[u][i] = fa[fa[u][i-1]][i-1]; //i的第2^j祖先就是i的第2^(j-1)祖先的第2^(j-1)祖先
for(int i=head[u];i!=-1;i=e[i].next) //每一个与u相连的节点i
{
int v = e[i].to;
if(v==f)
continue;
d[v] = d[u]+1; //更新深度
dis[v] = dis[u]+e[i].w;
fa[v][0] = u;
dfs(v,u);
}
}//深搜出各点的深度,存在d中
/*void bz()
{
for(int j=1;j<=N;j++)
for(int i=1;i<=n;i++)
fa[i][j] = fa[fa[i][j-1]][j-1];
}*/
int lca(int u,int v)
{
if(d[u]<d[v])
swap(u,v); //保证深度大的点为u,方便操作
int dc=d[u]-d[v]; //计算深度差
int i;
for(i=0;i<=N;i++) //值得注意的是,这里需要从零枚举
{ //此循环用于提到深度相同
if((1<<i)& dc)
u=fa[u][i];
}
if(u==v) //如果深度一样时,两个点相同,直接返回
return u;
for(i=N;i>=0;i--) //如果从小到大的话就有可能无法提到正确位置
{
if(fa[u][i]!=fa[v][i])//跳2^j步,如果不一样,就跳,否则不跳 如果相等有可能都为0 即超过根节点
{//在同一高度了,他们一起向上跳,跳他们不相同节点,当全都跳完之后fa[u][0]就是lca
u=fa[u][i];
v=fa[v][i];
}
}
u=fa[u][0]; //上述过程做完,两点都在LCA下一层,所以走一步即可
return u;
}
int main()
{
//std::ios::sync_with_stdio(false);
int u,v,w;
int x,y;
while(~scanf("%d%d%d",&n,&m,&q))
{
init();
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&u,&v,&w);
add(u,v,w);
add(v,u,w);
merge(u,v);
}
for(int i=1;i<=n;i++)
{
if(pre[i]==i)
{
d[i] = 1;
dfs(i,0);
}
}
//bz();
for(int i=0;i<q;i++)
{
scanf("%d%d",&x,&y);
if(find(x)!=find(y))
printf("Not connected\n");
else
printf("%d\n",dis[x]+dis[y]-2*dis[lca(x,y)]);
}
}
return 0;
}