HDU 5876 Sparse Graph (补图找最短路/BFS)

#include<map>
#include<queue>
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int maxn=1e6;
int head[maxn],le,n,m;
int dis[maxn];
int step[maxn];
map<int,int>e[maxn];

void bfs(int s)
{
queue<int>Q;
Q.push(s);
dis[s]=1;
step[s]=0;
int sum=1;
while(!Q.empty())
{
int now=Q.front();
Q.pop();
for(int i=1; i<=n ; i++)
{
if(e[now].count(i)==0)
{
if(dis[i]==0)
{
sum++;
Q.push(i);
dis[i]=1;
step[i]=step[now]+1;
}
}
if(sum==n)  break;
}
if(sum==n)  break;
}
for(int i=1; i<=n; i++)
{
if(s==i)    continue;
else if(i!=n)   printf("%d ",step[i]);
else printf("%d",step[i]);
}
printf("\n");
}

int main()
{
int t;
scanf("%d",&t);
while(t--)
{
le=1;
memset(head,-1,sizeof(head));
scanf("%d%d",&n,&m);
memset(dis,0,sizeof(dis));
for(int i=0; i<m; i++)
{
int a,b;    scanf("%d%d",&a,&b);
e[a]=1;  e[b][a]=1;
}
int s;
scanf("%d",&s);
bfs(s);
for(int i=1;i<=n;i++)   e[i].clear();
}
}