HDU 2066 一个人的旅行

#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
#define INF 0x3f3f3f3f
#define MAXN 1005
#define MAXM 1000005
using namespace std;

struct Node
{
int u,v,w;
int next;
}edge[MAXM];

int tol;
int head[MAXN];
int dist[MAXN];
int vis[MAXN];
int n,m,c,Max;

void init()
{
tol=0;
memset(head,-1,sizeof(head));
}

void addedge(int u,int v,int w)
{
edge[tol].u=u;
edge[tol].v=v;
edge[tol].w=w;
edge[tol].next=head[u];
head[u]=tol;
tol++;
}

void SPFA(int v0)
{
queue<int> q;
for(int i=0;i<=Max;i++)
{
dist[i]=INF;
vis[i]=0;
}
dist[v0]=0;
vis[v0]=1;
q.push(v0);
while(!q.empty())
{
int u=q.front();
q.pop();
vis[u]=0;
for(int i=head[u];i!=-1;i=edge[i].next)
{
int v=edge[i].v;
if(dist[v]>dist[u]+edge[i].w)
{
dist[v]=dist[u]+edge[i].w;
if(!vis[v])
{
q.push(v);
vis[v]=1;
}
}
}
}
}

int main()
{
while(~scanf("%d%d%d",&n,&m,&c)){
init();
int u,v,w;
for(int i=1;i<=n;i++){
scanf("%d%d%d",&u,&v,&w);
addedge(u,v,w);
addedge(v,u,w);
Max=max(Max,max(u,v));
}
for(int i=1;i<=m;i++){
scanf("%d",&v);
addedge(0,v,0);
}
SPFA(0);
int ans=INF;
for(int i=1;i<=c;i++){
scanf("%d",&v);
ans=min(ans,dist[v]);
}
printf("%d\n",ans);
}
return 0;
}