Hdu oj 2066 一个人的旅行

题目：点击打开链接

代码

#include<stdio.h>
#include<math.h>
#include<string.h>
#include<algorithm>
#define INF 0x3f3f3f3f
using namespace std;
int d[1005][1005];
int a1[1005],b1[1005];
int main()
{
	int m,n,p;
	while(~scanf("%d%d%d",&m,&n,&p))
	{
		int i,j,k;
		
		int a,b,c;
		int Min=INF;
		int N=0;
		memset(d,INF,sizeof(d));
		memset(a1,0,sizeof(a1));
		memset(b1,0,sizeof(b1));
		for(i=0;i<m;i++)
		{
			scanf("%d%d%d",&a,&b,&c);
			if(c<d[a][b])
			{
				d[a][b]=c;
				d[b][a]=c;
			}
			N=max(N,max(a,b));
		}
		for(i=0;i<n;i++)
		{
		    scanf("%d",&a);
		    a1[a]=1;
		}
		for(i=0;i<p;i++)
		{
			scanf("%d",&a);
			b1[a]=1;
		}
		for(k=1;k<=N;k++)
		    for(i=1;i<=N;i++)
		    if(d[i][k]!=INF)
		    {
		    	for(j=1;j<=N;j++)
		        {
		        	if(d[i][j]>d[i][k]+d[k][j])
		        	d[i][j]=d[i][k]+d[k][j];
		        	if(a1[i]==1&&b1[j]==1)
		        	Min=min(Min,d[i][j]);
		        }
		    }
		        
		printf("%d\n",Min);
	}
	return 0;
}