HDU 5876 2016 ACM/ICPC Asia Regional Dalian Online BFS+set

题目链接

解析

---

完全图删去m个边，求最短路。
set维护没有访问的点。

---

代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <queue>
#include <cmath>
#include <map>
#include <set>
#include <algorithm>
using namespace std;
typedef long long LL;
const int maxn = 200000+10;
const int maxe = 20000+10;
const long long inf = 0x3f3f3f3f;
typedef long long LL;
vector<int>g[maxn];
int dis[maxn];

void bfs(int S, int n) {
    queue<int>q;
    set<int>s, t;
    q.push(S);
    dis[S] = 0;
    for (int i=1; i<=n; i++)
        if (i != S)
            s.insert(i);

    while (!q.empty()) {
        int u = q.front();
        q.pop();
        for (int i=0; i<g[u].size(); i++) {
            int v = g[u][i];
            if (!s.count(v))
                continue;
            s.erase(v);
            t.insert(v);
        }
        for (set<int>::iterator it=s.begin(); it!=s.end(); it++) {
            int v = *it;
            dis[v] = dis[u]+1;
            q.push(v);
        }
        s.swap(t);
        t.clear();
    }
}

int main() {
    int T;
    scanf("%d", &T);
    while (T--) {
        int n, m;
        scanf("%d%d", &n, &m);
        for (int i=1; i<=n; i++)
            g[i].clear();
        for (int i=0; i<m; i++) {
            int u, v;
            scanf("%d%d", &u, &v);
            g[u].push_back(v);
            g[v].push_back(u);
        }
        int s;
        scanf("%d", &s);
        memset(dis, -1, sizeof(dis));
        bfs(s, n);
        for (int i=1; i<=n; i++) {
            if (i == s)
                continue;
            if (i == n)
                printf("%d\n", dis[i]);
            else
                printf("%d ", dis[i]);
        }
    }
    return 0;
}