Description
给出一个n个点m条边的无向图,边权均为1,给出一起点s,求其补图任一点到s的最短路长度
Input
第一行一整数T表示用例组数,每组用例首先输入两整数n和m表示点数和边数,之后m行每行两个整数u和v表示u和v之间有一条无向边,最后一整数s表示起点
(1<=T < 35,2<=n<=200000,0<=m<=20000,1<=u,v,s<=n)
Output
对于每组用例,输出n-1个整数表示补图中其他点到s的最短路长度
Sample Input
1
2 0
1
Sample Output
1
Solution
直接在补图上做最短路显然不行,因为边权为1考虑bfs,每次扩展的点是S\A,S为当前尚为扩展的点集,A为当前点的邻接点集,扩展完后将A\S赋给S,可以用两个set存S集合和A集合,那么每条边的两个端点至多被加入A集中一次,从S集中被删除一次,时间复杂度O(Tmlogn)
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
#include<set>
using namespace std;
typedef long long ll;
#define maxn 222222
vector<int>g[maxn];
int T,n,m,s,dis[maxn];
void bfs(int s)
{
queue<int>que;
set<int>s1,s2;
set<int>::iterator it;
for(int i=1;i<=n;i++)
if(i!=s)s1.insert(i);
que.push(s);
memset(dis,-1,sizeof(dis));
dis[s]=0;
while(!que.empty())
{
int u=que.front();
que.pop();
for(int i=0;i<g[u].size();i++)
{
int v=g[u][i];
if(s1.find(v)==s1.end())continue;
s1.erase(v),s2.insert(v);
}
for(it=s1.begin();it!=s1.end();it++)
{
dis[*it]=dis[u]+1;
que.push(*it);
}
s1.swap(s2);
s2.clear();
}
}
int main()
{
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)g[i].clear();
while(m--)
{
int u,v;
scanf("%d%d",&u,&v);
g[u].push_back(v),g[v].push_back(u);
}
scanf("%d",&s);
bfs(s);
int res=0;
for(int i=1;i<=n;i++)
if(i!=s)
res++,printf("%d%c",dis[i],res==n-1?'\n':' ');
}
return 0;
}