Hdu 2066 一个人的旅行

Hdu 2066 一个人的旅行(floyd算法)

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=2066


解题思路:因为起点城镇和终点城镇的极限数据就是n次迪杰斯特拉，即O(n^3),所以不然直接干脆用弗洛伊德算法处理，写起来还简单.
代码如下:

#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f
#define MAXN 1005
int t,s,d;
int v[MAXN][MAXN],a[MAXN],b[MAXN];
int main(){
    while(scanf("%d %d %d",&t,&s,&d)!=EOF){
        for(int i=1;i<=1000;i++){//因为不知到出现城市数字的范围，就按题目给出的最大来算
            for(int j=1;j<=1000;j++){
                if(i==j)v[i][j]=0;
                else v[i][j]=inf;
            }
        }
        int m=0;//记录出现地点的最大值
        for(int i=0;i<t;i++){
            int c,f,e;
            scanf("%d %d %d",&c,&e,&f);
            v[c][e]=v[e][c]=f>v[c][e]?v[c][e]:f;
            m=m>c?m:c;
            m=m>e?m:e;
        }
        for(int i=1;i<=s;i++){
            scanf("%d",&a[i]);
            m=m>a[i]?m:a[i];
        }
        for(int i=1;i<=d;i++){
            scanf("%d",&b[i]);
            m=m>b[i]?m:b[i];
        }
        for(int k=1;k<=m;k++){//弗洛伊德算法
            for(int i=1;i<=m;i++){
                if(v[k][i]==inf)continue;
                for(int j=1;j<=m;j++){
                    v[i][j]=v[i][j]>v[i][k]+v[k][j]?v[i][k]+v[k][j]:v[i][j];
                }
            }
        }
        int mn=inf;
        for(int i=1;i<=s;i++){//找最小值
            for(int j=1;j<=d;j++){
                mn=mn>v[a[i]][b[j]]?v[a[i]][b[j]]:mn;
            }
        }
        printf("%d\n",mn);
    }
    return 0;
}