题目:
给一个图 G,称 G 的补图为 H,给一个点 S ,求 H 中 S 与剩下 N - 1个点的最短路。
分析:
因为题目点数200000,而原图边数只有20000,所以补图的边会非常多,所以跑最短路是肯定不行的。
感觉补图是个重要条件,因为补图里会有一大部分点和s点直接相连,得利用这个条件。看题解用bfs,主要思想就是找出那些补图里可以直接相连的点。
bfs是每次用一个“当前点”往外扩展,每个点只访问一次。所以用一个集合s1表示原图里不能到达当前点的点,也就是补图里直接连边的点。s2用来过度,存当前点可以到达的点,也就是这些点答案还不能确定,下次从这些点里再找补图里能直接到达的。
代码:
#include <iostream>
#include <algorithm>
#include <queue>
#include <stack>
#include <vector>
#include <map>
#include <set>
#include <cmath>
#include <cstdlib>
#include <cstring>
#include <cstdio>
using namespace std;
#define ms(a,b) memset(a,b,sizeof(a))
#define lson rt*2,l,(l+r)/2
#define rson rt*2+1,(l+r)/2+1,r
typedef unsigned long long ull;
typedef long long ll;
const int MAXN=2e5+5;
const double EPS=1e-8;
const int INF=0x3f3f3f3f;
const int MOD = 1e9+7;
vector<int>g[MAXN];
int n,m,st,dis[MAXN],vis[MAXN];
void init(){
for(int i=0;i<MAXN;i++){
g[i].clear();
}
}
void bfs(){
ms(vis,0);
ms(dis,-1);
queue<int> q;
set<int> s1,s2;
for(int i=1;i<=n;i++){
if(i!=st) s1.insert(i);
}
vis[st] = 1;
dis[st] = 0;
q.push(st);
while(!q.empty()){
int u = q.front(); q.pop();
for(int i=0;i<(int)g[u].size();i++){
int v = g[u][i];
if(!s1.count(v)) continue;
s1.erase(v);
s2.insert(v);
}
for(auto it = s1.begin();it!=s1.end();it++){
q.push(*it);
dis[*it] = dis[u] + 1;
}
s1.swap(s2);
s2.clear();
}
}
int main(){
int T;
scanf("%d",&T);
while(T--){
init();
scanf("%d%d",&n,&m);
while(m--){
int u,v;
scanf("%d%d",&u,&v);
g[u].push_back(v);
g[v].push_back(u);
}
scanf("%d",&st);
bfs();
if(st==1){
for(int i=2;i<=n;i++){
printf("%d%c",dis[i]," \n"[i==n]);
}
}else if(st==n){
for(int i=1;i<=n-1;i++){
printf("%d%c",dis[i]," \n"[i==n-1]);
}
}else{
for(int i=1;i<=n;i++){
if(st!=i) printf("%d%c",dis[i]," \n"[i==n]);
}
}
}
return 0;
}