HDU 2680(spfa)

題意:有多個起點,求到終點的最短路,一個個求的話很明顯會超時。有兩種寫法:

一,構建一個超級源點,與起點相連,距離爲0,從0開始搜。然後套模板即可。

二,構建反向圖,以終點爲起點開始搜。

下面是我用第一種思路套的spfa 的模板。。寫的有點醜

#include<stdio.h>
#include<string.h>
#include<queue>
using namespace std;
const int N=1100;
const int M=22000;
const int INF=0x3f3f3f3f;
int d[N],p[N],cnt[N],n,js;
vector<int> g[N];
struct node
{
    int v,w;
}e[M];
void add(int u,int v,int w)
{
    e[js].v=v,e[js].w=w;
    g[u].push_back(js++);
}
void init()
{
    for(int i=0;i<=n;i++)
    {
        d[i]=INF;
        g[i].clear();
    }
    memset(p,0,sizeof(p));
    memset(cnt,0,sizeof(cnt));
    js=0;
}
int spfa(int s)
{
    queue<int> q;
    int u,v,w,i;
    q.push(s),d[s]=0,p[s]=1,cnt[s]=1;
    while(!q.empty())
    {
        p[u=q.front()]=0;
        q.pop();
        for(i=0;i<g[u].size();i++)
        {
            v=e[g[u][i]].v,w=e[g[u][i]].w;
            if(d[u]+w<d[v])
            {
                d[v]=d[u]+w;
                if(!p[v])
                {
                    if(++cnt[v]>n)
                        return 0;
                    p[v]=1;
                    q.push(v);
                }
            }
        }
    }
    return 1;
}
int main()
{
    int m,i,u,v,w,t;
    while(scanf("%d%d%d",&n,&m,&t)!=EOF)
    {
        n++;
        init();
        while(m--)
        {
            scanf("%d%d%d",&u,&v,&w);
            add(u,v,w);
        }
        scanf("%d",&m);
        while(m--)
        {
            scanf("%d",&v);
            add(0,v,0);
        }
        if(spfa(0))
        {
            if(d[t]==INF)
                printf("-1\n");
            else
                printf("%d\n",d[t]);
        }
        else
            printf("-1\n");
    }
    return 0;
}



發佈了30 篇原創文章 · 獲贊 1 · 訪問量 8183
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章