抖音APP源碼hdu 5876 Sparse Graph 完全圖補圖最短路

利用不能走的圖的邊(邊少) 首先把所有點入隊,然後每走一個點,把沒有邊的點得到距離(因爲其補圖必定能到),然後繼續入隊
然後不斷更新剩下的點 時間複雜度 是o(n+m)
一共走n個點和m條邊

#include <bits/stdc++.h>
using namespace std;
int n,m;
const int N = 201000;
vector<int> v[N];
int d[N],s;
set<int> st1,st2;
const int inf = 0x3f3f3f3f;

void init()
{
    for(int i=1;i<=n;i++)
    {
        d[i]=inf;
        v[i].clear();
    }
}

queue<int> Q;

void bfs()
{
    st1.clear();
    for(int i=1;i<=n;i++)
    {
        if(i!=s) st1.insert(i);
    }
    while(!Q.empty()) Q.pop();
    d[s]=0;
    Q.push(s);
    while(!Q.empty())
    {
        int top=Q.front();
        Q.pop();
        for(int i=0;i<v[top].size();i++){
            if(st1.count(v[top][i]))
            {
            st1.erase(v[top][i]);
            st2.insert(v[top][i]);
            }
        }
        for(set<int>::iterator it = st1.begin();it!=st1.end();it++) 
        {
            d[*it]=d[top]+1;
            Q.push(*it);
        }
        st1.swap(st2);
        st2.clear();
    }
}

int main()
{
    int t;
    while(scanf("%d",&t)!=EOF)
    {
        while(t--)
        {
        scanf("%d%d",&n,&m);

        init();
        for(int i=1;i<=m;i++)
        {
            int c,d;
            scanf("%d%d",&c,&d);
            v[c].push_back(d);
            v[d].push_back(c);
        }
        scanf("%d",&s);
        bfs();
        int f=0;
        for(int i=1;i<=n;i++)
        {
            if(i==s) continue;
            if(f) printf(" ");
            if(d[i]==inf) printf("-1");
            else printf("%d",d[i] );
            f=1;
        }
        printf("\n");
        }
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章