hdu5876Sparse Graph

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

题意：给定一个无向图的m条边，求其补图以点s为起点的最短路。

分析：用一个队列维护用于更新的点和一个set维护哪些点还没有被更新，每次将于当前点直接相连的边丢进没有被更新的set中即可。

代码：

#include<map>
#include<set>
#include<cmath>
#include<queue>
#include<bitset>
#include<math.h>
#include<vector>
#include<string>
#include<stdio.h>
#include<cstring>
#include<iostream>
#include<algorithm>
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
const int N=200010;
const int mod=1000000007;
const int MOD1=1000000007;
const int MOD2=1000000009;
const double EPS=0.00000001;
typedef long long ll;
const ll MOD=1000000007;
const int INF=1000000010;
const ll MAX=1ll<<55;
const double eps=1e-5;
const double inf=~0u>>1;
const double pi=acos(-1.0);
typedef double db;
typedef unsigned int uint;
typedef unsigned long long ull;
queue<int>D;
set<int>S,B;
vector<int>Q[N];
int ans[N];
int main()
{
    int a,b,i,n,m,s,T,now;
    scanf("%d", &T);
    while (T--) {
        scanf("%d%d", &n, &m);
        for (i=1;i<=n;i++) Q[i].clear();
        for (i=1;i<=m;i++) {
            scanf("%d%d", &a, &b);
            Q[a].push_back(b);Q[b].push_back(a);
        }
        scanf("%d", &s);
        for (i=1;i<=n;i++) ans[i]=-1;
        ans[s]=0;D.push(s);S.clear();
        for (i=1;i<=n;i++)
        if (i!=s) S.insert(i);
        while (!D.empty()) {
            now=D.front();D.pop();
            for (i=0;i<Q[now].size();i++)
            if (ans[Q[now][i]]==-1) {
                S.erase(Q[now][i]);B.insert(Q[now][i]);
            }
            for (set<int>::iterator it=S.begin();it!=S.end();it++) {
                ans[*it]=ans[now]+1;D.push(*it);
            }
            S.clear();
            for (set<int>::iterator it=B.begin();it!=B.end();it++) S.insert(*it);
            B.clear();
        }
        if (s==n) {
            for (i=1;i<n-1;i++) printf("%d ", ans[i]);
            printf("%d\n", ans[n-1]);
        } else {
            for (i=1;i<n;i++)
            if (i!=s) printf("%d ", ans[i]);
            printf("%d\n", ans[n]);
        }
    }
    return 0;
}