人活着系列之芳姐和芳姐的猪---2929

人活着系列之芳姐和芳姐的猪

Time Limit: 1000MS  Memory Limit: 65536KB

Problem Description

百年来，人活着是为了什么这个问题一直萦绕在人的脑海里，也一直困扰着人的思想。人活着就是活着了，为活着本身而活着，而不是为活着之外的任何事物而活着的。正因为活着，所以活着。对，是有点莫明其妙，但也是一句最受用的话。

芳姐特别喜欢猪，所以，她特意养了n头猪，建了m个猪圈，顺便在m个猪圈间修了k条无向边，每条边有都有起点u，终点v，距离w。每头猪呆在一个特定的猪圈，有一个问题一直困扰着芳姐，那就是喂猪.....芳姐和猪们约定好，每天去一个固定猪圈去吃饭，芳姐为了不累着她可爱的猪们，想知道所有的猪吃饭走的最短路程是多少？

Input

 第一行 : 三个数 ,猪的个数 n(1<=n<=350)，猪圈个数 m（ 2<=m<=600），猪圈间道路数 k(1<=k<=1200).（猪的编号为 1..N, 猪圈的编号为 1..m)

第二行到第N+1行: 1到N头猪所在的猪圈号.

第n+2行到第n+k+1行： 每行有三个数：相连的猪圈u、v，两猪圈间距离（1<=w<=255)

注：有的猪圈可能是空的，也可能有多头猪，保证m个猪圈连通。

Output

 

Example Input

3 4 5
2
3
4
1 2 1
1 3 5
2 3 7
2 4 3
3 4 5

Example Output

8

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#define inf 0x3f3f3f3f
//Floyd-Warshall解决'多源最短路径问题'
int a[605][605],count[400];

void Floyd(int m)
{
    int i,l,j;
    for(l=1; l<=m; l++)
    {
        for(i=1; i<=m; i++)
        {
            for(j=1; j<=m; j++)
            {
                if(a[i][j]>a[i][l]+a[l][j])
                    a[i][j]=a[i][l]+a[l][j];
            }
        }
    }
}
int main()
{
    int n,m,k,u,v,w,i,j,min=999999,sum=0;
    scanf("%d%d%d",&n,&m,&k);
    memset(a,inf,sizeof(a));
    memset(count,0,sizeof(count));
    for(i=1; i<=n; i++)
    {
        scanf("%d",&count[i]);
    }
    for(i=1; i<=m; i++)
        for(j=1; j<=m; j++)
            if(i==j)  a[i][j]=0;
    while(k--)
    {
        scanf("%d%d%d",&u,&v,&w);
        a[u][v]=w;
        a[v][u]=w;
    }
    Floyd(m);
    for(i=1;i<=m;i++)
    {
        sum=0;
        for(j=1;j<=n;j++)
        {
           sum+=a[count[j]][i];
        }
        if(sum<min)
        min=sum;
    }
    printf("%d\n",min);
    return 0;
}