hdu 5876 - Sparse Graph(2016大連網絡賽) bfs

題意:補圖就是和原圖聯通狀態相反的圖,兩個點相鄰那麼在補圖裏他們不相鄰給出一個圖的補圖,求出s到其他點的最短距離。

用set存放還沒有走到過的點,然後用bfs依次求到每個點的最短距離,到達一個點後就把這個點從set中刪掉,這樣可以使bfs的複雜度非常接近o(n)。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
#include <set>
using namespace std;
const int MAXN = 200010;
set<int> edge[MAXN], ss;
set<int>::iterator p, p1, p2;
queue<int> q;
int dist[MAXN];
int main() {
    int t, n, m, s, i, j;
    scanf("%d", &t);
    while(t--) {
        ss.clear();
        scanf("%d%d", &n, &m);
        for(i = 1; i <= n; i++) {
            edge[i].clear();
        }
        int a, b;
        for(i = 0; i < m; i++) {
            scanf("%d%d", &a, &b);
            edge[a].insert(b);
            edge[b].insert(a);
        }
        for(i = 1; i <= n; i++) {
            ss.insert(i);
        }
        memset(dist, -1, sizeof(dist));
        scanf("%d", &s);
        int xx = s;
        q.push(s);
        dist[s] = 0;
        ss.erase(s);
//        for(p = ss.begin(); p != ss.end(); p++)
//            cout << *p << endl;
        while(q.size()) {
            s = q.front();
            q.pop();
//            cout << "##  " << s << endl;
            for(p = ss.begin(); p != ss.end();) {
//                cout << "SS  " << edge[s].size() << endl;
                if(!edge[s].count(*p)) {
                    q.push(*p);
                    dist[*p] = dist[s] + 1;
                    p1 = p++;
                    ss.erase(p1);
                }
                else p++;
            }
        }
        int is = 0;
        for(i = 1; i <= n; i++) {
            if(i == xx) continue;
            if(is)
                printf(" ");
            else is = 1;
            printf("%d", dist[i]);
        }
        printf("\n");
    }
    return 0;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章