补图 + bfs

题目链接

http://acm.hdu.edu.cn/showproblem.php?pid=5876

题意：n 个点的无向完全图中删除 m 条边，问点 s 到其他点的最短路长度。

题解：

补图上的 BFS 是非常经典的问题。一般的做法是用链表（或者偷懒用 std::set）维护还没 BFS 过的点。当要扩展点 u 的时候，遍历一次还没访问过的点 v，如果 uv 没边，那么将 v 入队。否则将 v 留在未扩展点中。

很明显，后者只会发生 m 次，前者只会发生 n 次，所以复杂度是  O(n + m)O(n+m) .

#include <cstdio>
#include <algorithm>
#include <iostream>
#include <cstring>
#include <string>
#include <cmath>
#include <queue>
#include <vector>
#include <set>
using namespace std;
#define N 200010
#define INF 0x3f3f3f3f

struct node
{
    int v, nxt;
}edge[N];
int head[N];
int d[N], tot;

void add(int u, int v)
{
    edge[tot].v = v; edge[tot].nxt = head[u]; head[u] = tot++;
    edge[tot].v = u; edge[tot].nxt = head[v]; head[v] = tot++;
}

void bfs(int st, int n)
{
    queue<int> que;
    d[st] = 0;
    que.push(st);
    set<int> s1, s2;
    for(int i = 1; i <= n; i++) {
        if(i != st) s1.insert(i);
    }
    while(!que.empty()) {
        int u = que.front(); que.pop();
        for(int k = head[u]; ~k; k = edge[k].nxt) {
            int v = edge[k].v;
            if(!s1.count(v)) continue;
            s1.erase(v); //补图中当前不能走到的并且还未更新过的
            s2.insert(v); //补图中走不到的还要扩展的
        }
        for(set<int>:: iterator it = s1.begin(); it != s1.end(); it++) {
            d[*it] = d[u] + 1;
            que.push(*it);
        }
        s1.swap(s2); //还没更新过的
        s2.clear();
    }
}

int main()
{
    int t;
    scanf("%d", &t);
    while(t--) {
        memset(head, -1, sizeof(head));
        memset(d, INF, sizeof(INF));
        int n, m;
        scanf("%d%d", &n, &m);
        for(int i = 0; i < m; i++) {
            int u, v;
            scanf("%d%d", &u, &v);
            add(u, v);
        }
        int st;
        scanf("%d", &st);
        bfs(st, n);
        bool f = 0;
        for(int i = 1; i <= n; i++) {
            if(i != st) {
                if(f) printf(" ");
                f = 1;
                if(d[i] == INF) printf("-1");
                else printf("%d", d[i]);
            }
        }
        puts("");
    }

    return 0;
}