hdu 2680 Choose the best route

http://acm.hdu.edu.cn/showproblem.php?pid=2680

这道题是有向图。把全部可以出发的点进队列spfa就可以。

#include <cstdio>
#include <vector>
#include <queue>
#include <cstring>
#include <algorithm>
#define maxn 1001
using namespace std;
const int inf=1<<28;

int n,m,s;
int p,q,t;
int g[maxn][maxn];
int dis[maxn];
int c[maxn];
int w;
bool vis[maxn];
void dijstra(int src)
{
    memset(vis,false,sizeof(vis));
    for(int i=1; i<=n; i++) dis[i]=inf;
    dis[src]=0;
    for(int i=1; i<=n; i++)
    {
        int x,m=inf;
        for(int y=1; y<=n; y++) if(!vis[y]&&dis[y]<m) m=dis[x=y];
        vis[x]=true;
        for(int y=1; y<=n; y++) dis[y]=min(dis[y],dis[x]+g[x][y]);
    }
}

void spfa()
{
    queue<int>q;
    memset(vis,false,sizeof(vis));
    for(int i=1; i<=n; i++) dis[i]=inf;
    for(int i=0; i<w; i++)
    {
        dis[c[i]]=0;
        vis[c[i]]=true;
        q.push(c[i]);
    }
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        vis[u]=false;
        for(int i=1; i<=n; i++)
        {
            if(dis[i]>dis[u]+g[u][i]&&g[u][i]!=inf)
            {
                dis[i]=dis[u]+g[u][i];
                if(!vis[i])
                {
                    vis[i]=true;
                    q.push(i);
                }
            }
        }
    }
}

int main()
{
    while(scanf("%d%d%d",&n,&m,&s)!=EOF)
    {
        for(int i=1; i<=n; i++)
        {
            for(int j=1; j<=n; j++)
            {
                g[i][j]=inf;
            }
        }
        for(int i=0; i<m; i++)
        {
            scanf("%d%d%d",&p,&q,&t);
            g[p][q]=min(g[p][q],t);
        }
        scanf("%d",&w);
        for(int i=0; i<w; i++)
        {
            scanf("%d",&c[i]);
        }
        spfa();
        if(dis[s]==inf)
            printf("-1
");
        else printf("%d
",dis[s]);
    }
    return 0;
}