poj 1603 Risk(最短路floyd)

边权为 1 的最短路,floyd模板直接过。


有几个debug,错了几次详见代码。


代码:

#include<stdio.h>
#include<algorithm>

using namespace std;
const int MaxN = 25;
const int INF = 0x3f3f3f3f;

int g[MaxN][MaxN];

void floyd()
{
    for(int k = 1; k <= 20; k++)
        for(int i = 1; i <= 20; i++)
            for(int j = 1; j <= 20; j++)
                g[i][j] = min(g[i][j], g[i][k] + g[k][j]);
}

int main()
{
    #ifdef LOCAL
    freopen("in.txt", "r", stdin);
    #endif // LOCAL
    int num, v;
    int count = 1;
    while(scanf("%d",&num)!=EOF)
    {
        //int count = 1;//debug
        for(int i = 1; i <= 20; i++)
        {
            for(int j = 1; j <= 20; j++)
            {
                if(i == j)
                    g[i][j] = 0;
                else
                    g[i][j] = INF;
            }
        }
        int k = 1;
        for(int i = 0; i < num; i++)
        {
            scanf("%d", &v);
            g[k][v] = 1;
            g[v][k] = 1;//debug
        }
        for(int j = 2; j <= 19; j++)
        {
            k++;
            scanf("%d", &num);
            for(int i = 0; i < num; i++)
            {
                scanf("%d", &v);
                g[k][v] = 1;
                g[v][k] = 1;//debug
            }
        }
        floyd();
        printf("Test Set #%d\n", count);
        count ++;
        int kk, go, to;
        scanf("%d", &kk);
        for(int i = 0; i < kk; i++)
        {
            scanf("%d %d", &go, &to);
            printf("%d to %d: %d\n", go, to, g[go][to]);
        }
        printf("\n");//debug
    }
    return 0;
}


你可能感兴趣的:(poj 1603 Risk(最短路floyd))