题目
题意:
给出n个点(n<=200000),m(m<=20000)条边的图,求补图单源最短路。
解法:
按照正常的做法,补图的边数太多会超时。有一个性质是,如果当前拓展点x和某些未拓展的点y没有边相连,那么这些未访问的点y的d[y]=d[x]+1,将这个d再放入队列,依次进行。整个·过程需要维护一个集合,存放尚未拓展的点。
代码:
#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<queue>
#include<set>
using namespace std;
#define all(x) (x).begin(), (x).end()
#define for0(a, n) for (int (a) = 0; (a) < (n); (a)++)
#define for1(a, n) for (int (a) = 1; (a) <= (n); (a)++)
#define mes(a,x,s) memset(a,x,(s)*sizeof a[0])
#define mem(a,x) memset(a,x,sizeof a)
typedef long long ll;
typedef pair<int, int> pii;
const int INF =0x3f3f3f3f;
const int maxn= 200000 ;
int n,m;
int x,y;
vector<int >G[maxn+10];
int S,d[maxn+5];
bool done[maxn+5];
void solve()
{
queue<int >q;q.push(S);
mes(d,0x3f,n+1);d[S]=0;
mes(done,0,n+1);
set<int >se[2];
for1(i,n) if(i!=S) se[0].insert(i);
int now=1;
while(!q.empty())
{
int x=q.front();q.pop();
if(done[x]) continue;done[x]=1;
now^=1;
int siz=G[x].size();
for0(i,siz)
{
int y=G[x][i];
if(!se[now].count(y)) continue;
se[now].erase(y);
se[now^1].insert(y);
}
for(set<int>::iterator it=se[now].begin();it!=se[now].end();it++)
{
int y=*it;
d[y]=d[x]+1;
q.push(y);
}
se[now].clear();
}
int cnt=0;
for1(i,n) if(i!=S)
{
cnt++;
int ret=d[i]==INF?-1:d[i];
printf("%d%c",ret,cnt==n-1?'\n':' ');
}
}
int main()
{
std::ios::sync_with_stdio(false);
int T;cin>>T;
while(T--)
{
cin>>n>>m;
for1(i,n) G[i].clear();
for1(i,m)
{
cin>>x>>y;
G[x].push_back(y);
G[y].push_back(x);
}
cin>>S;
solve();
}
return 0;
}