HDU2066,dijkstra算法,链式前向星

详见代码:

/*************************************************************************
    > File Name: main.cpp
    > Author:Eagles 
    > Mail:None 
    > Created Time: 2018年09月06日 星期四 19时11分20秒
    > Description:HDU2066 
 ************************************************************************/

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
using namespace std;
#define N 1000
struct node
{
    int to;
    int val;
    int nex;
}E[100000];//边

int dis[N][N];相邻点到各个点的最短距离
bool vis[N][N];
int want[N];//想去的城市
int head[N];
int lin[N];//相邻的城市
int t,s,d;
int cnt,num;//num是城市的个数

void addEdge(int a, int b, int val)
{
    E[cnt].val=val;
    E[cnt].to=b;
    E[cnt].nex=head[a];
    head[a]=cnt++;

    E[cnt].val=val;
    E[cnt].to=a;
    E[cnt].nex=head[b];
    head[b]=cnt++;
}

void init()
{

    memset(dis,0x3f3f3f3f,sizeof(dis));
    memset(vis,false,sizeof(vis));
    memset(head,-1,sizeof(head));
    cnt=0;
    num=0;

    for (int i=0; i<t; i++)
    {
        int a,b,val;
        scanf("%d%d%d",&a,&b,&val);
        num=max(num,max(a,b));
        addEdge(a-1,b-1,val);//下标从0开始
    }

    for (int i=0; i<s; i++)
    {
        int a;
        scanf("%d",&a);
        num=max(a,num);
        a--;
        lin[i]=a;
    }

    for (int i=0; i<d; i++)
    {
        int a;
        scanf("%d",&a);
        num=max(a,num);
        a--;
        want[i]=a;
    }

    for (int i=0; i<s; i++)
    {
        int k=head[lin[i]];

        while (k != -1)
        {
            if (dis[lin[i]][E[k].to]>E[k].val)
                dis[lin[i]][E[k].to]=E[k].val;
            k=E[k].nex;
        }

        dis[lin[i]][lin[i]]=0;

        vis[lin[i]][lin[i]]=true;
    }
}

void dijkstra()
{
    for (int i=0; i<s; i++)//对每一个相邻的城市进行dijkstra
    {
        for (int j=0; j<num-1; j++)
        {
            int minn=152561613;
            int tmp;

            for (int k=0; k<num; k++)
            {
                if (!vis[lin[i]][k]&&dis[lin[i]][k]<minn)
                {
                    minn=dis[lin[i]][k];
                    tmp=k;
                }
            }

            vis[lin[i]][tmp]=true;

            int don=head[tmp];

            while (don != -1)
            {
                if (!vis[lin[i]][E[don].to]&&dis[lin[i]][E[don].to]>dis[lin[i]][tmp]+E[don].val)
                dis[lin[i]][E[don].to]=dis[lin[i]][tmp]+E[don].val;

                don=E[don].nex;
            }
        }
    }
}

void ans()//获得答案
{
    int ans=15651613;//保证足够大就行

    for (int i=0; i<s; i++)
    {
        for (int j=0; j<d; j++)
        {
            ans=min(ans,dis[lin[i]][want[j]]);
        }
    }

    printf("%d\n",ans);
}

int main()
{

    while (~scanf("%d%d%d",&t,&s,&d))
    {
        init();
        dijkstra();
        ans();
    }
    return 0;
}