题意:给定一个无向图的m条边,求其补图以点s为起点的最短路。
分析:用一个队列维护用于更新的点和一个set维护哪些点还没有被更新,每次将于当前点直接相连的边丢进没有被更新的set中即可。
代码:
#include<map>
#include<set>
#include<cmath>
#include<queue>
#include<bitset>
#include<math.h>
#include<vector>
#include<string>
#include<stdio.h>
#include<cstring>
#include<iostream>
#include<algorithm>
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
const int N=200010;
const int mod=1000000007;
const int MOD1=1000000007;
const int MOD2=1000000009;
const double EPS=0.00000001;
typedef long long ll;
const ll MOD=1000000007;
const int INF=1000000010;
const ll MAX=1ll<<55;
const double eps=1e-5;
const double inf=~0u>>1;
const double pi=acos(-1.0);
typedef double db;
typedef unsigned int uint;
typedef unsigned long long ull;
queue<int>D;
set<int>S,B;
vector<int>Q[N];
int ans[N];
int main()
{
int a,b,i,n,m,s,T,now;
scanf("%d", &T);
while (T--) {
scanf("%d%d", &n, &m);
for (i=1;i<=n;i++) Q[i].clear();
for (i=1;i<=m;i++) {
scanf("%d%d", &a, &b);
Q[a].push_back(b);Q[b].push_back(a);
}
scanf("%d", &s);
for (i=1;i<=n;i++) ans[i]=-1;
ans[s]=0;D.push(s);S.clear();
for (i=1;i<=n;i++)
if (i!=s) S.insert(i);
while (!D.empty()) {
now=D.front();D.pop();
for (i=0;i<Q[now].size();i++)
if (ans[Q[now][i]]==-1) {
S.erase(Q[now][i]);B.insert(Q[now][i]);
}
for (set<int>::iterator it=S.begin();it!=S.end();it++) {
ans[*it]=ans[now]+1;D.push(*it);
}
S.clear();
for (set<int>::iterator it=B.begin();it!=B.end();it++) S.insert(*it);
B.clear();
}
if (s==n) {
for (i=1;i<n-1;i++) printf("%d ", ans[i]);
printf("%d\n", ans[n-1]);
} else {
for (i=1;i<n;i++)
if (i!=s) printf("%d ", ans[i]);
printf("%d\n", ans[n]);
}
}
return 0;
}