BFS(环) - Subway - CodeForces 131D
题意:
给定一个n个点,n条边的无向图,
整个图由一个环和多条链组成,
输出所有的点到环的距离。(环上的点到环的距离为0,链上的边权为1.)
输入:
首行包括一个正整数n,
接着n行表示n条边,每行两个正整数ui和vi,表示点ui和vi相连。
输出:
n个正整数,依次表示n个点到环的距离。
Examples
Input
4
1 3
4 3
4 2
1 2
Output
0 0 0 0
Input
6
1 2
3 4
6 4
2 3
1 3
3 5
Output
0 0 0 1 1 2
数据范围:
3 ≤ n ≤ 3000,1 ≤ ui, vi ≤ n
分析:
由于环上的点到环的距离都为0,我们可以把环看作一个整体,环上的点之间的边权为0。
那么我们以环上的点为源点跑最短路即可。
如何标记环上的点?
我们可以从所有度为1的点(链的末端)开始进入搜索,d表示点的度数。
①、当搜到d>2的点时,我们就将这个点的度数减1。
②、当搜到度数为2的点时,就将其标记为链上的点。
下证为何上述方法能够标记掉链上的点:
从链的末端扩展,有两种情况:
①、扩展到的点的度数d=2,该点必是链上的点,直接标记掉。
②、扩展到的点的度数d>2:
Ⅰ、该点是多条链的交点:将这个点的度数减1,表示我们已经搜索完了链的一条分支。
此时不会将该点入队,不会从该点继续向下扩展。
当这条链的所有分支都被搜索过后,整个链就会变成一条没有分支的单链,
从链的末端进入搜索,就会将整个链上的点都标记掉。
Ⅱ、该点的环上的点:同样地,将该点的度数减1后,并不会将该点入队,即不会从该点继续向下扩展。
由于环上点的度数恒大于等于2,所以直到所有与该点连接的链都被标记完,该点也不会入队。
这就保证了环内部的点不会被扩展到。
通过上述方法,未被标记的点就是环上的点,从环上的点向外跑bfs就能够得到答案。
具体细节:
①、先将度数为1的点入队,向外扩展,将所有链上的点标记出来。
②、将未被标记的环上的点入队,向外扩展,跑bfs。
代码:
#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=3010, M=2*N;
int n;
int e[M],ne[M],h[N],idx;
bool st[N];
int d[N];
int ans[N];
void add(int a,int b)
{
e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
void Find_loop()
{
queue<int> Q;
for(int i=1;i<=n;i++)
if(d[i]==1)
Q.push(i), st[i]=true;
while(Q.size())
{
int u=Q.front();
Q.pop();
for(int i=h[u];~i;i=ne[i])
{
int j=e[i];
if(st[j]) continue;
if(d[j]==2) Q.push(j), st[j]=true;
else if(d[j]>=3) d[j]--;
}
}
}
void bfs()
{
queue<int> Q;
for(int i=1;i<=n;i++)
if(!st[i])
Q.push(i);
while(Q.size())
{
int u=Q.front();
Q.pop();
for(int i=h[u];~i;i=ne[i])
{
int j=e[i];
if(!st[j]) continue;
st[j]=false;
Q.push(j);
ans[j]=ans[u]+1;
}
}
}
int main()
{
cin>>n;
memset(h,-1,sizeof h);
int a,b;
for(int i=0;i<n;i++)
{
cin>>a>>b;
add(a,b), add(b,a);
d[a]++, d[b]++;
}
Find_loop();
bfs();
for(int i=1;i<=n;i++) cout<<ans[i]<<' ';
return 0;
}