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

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

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

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

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

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

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

8

#include <stdio.h>
#include <stdlib.h>
#define INF 0x3f3f3f3f
int map[605][605];
int c[605];
void floyd(int n);
int main()
{int n,m,k,i,u,v,w,num;while(scanf("%d%d%d",&n,&m,&k) != EOF){memset(map,INF,sizeof(map));memset(c,0,sizeof(c));for(i = 1;i <= m;i++){map[i][i] = 0;}for(i = 0;i < n;i++){scanf("%d",&num);c[num]++;}for(i = 0;i < k;i++){scanf("%d%d%d",&u,&v,&w);map[u][v] = map[v][u] = w;}floyd(m);}
}
void floyd(int n)
{int i,j,k,sum,su;for(k = 1;k <= n;k++){for(i = 1;i <=n;i++){for(j = 1;j <=n;j++){if(map[i][j] > map[i][k]+map[k][j]){map[i][j] = map[i][k] + map[k][j];}}}}sum = INF;for(i = 1;i <= n;i++){su = 0;for(j = 1;j <= n;j++){su += map[j][i]*c[j];}if(sum > su)sum = su;}printf("%d\n",sum);
}