洛谷P1364 医院设置

LITTLESUN的第一道图论,撒花~~

题目链接

这道题是Floyd的板子题
注意对于矩阵图的初始值赋值要全部赋值成最大值
十六进制的最大值表示方式是0x3f3f3f3f

memset(G,0x3f,sizeof(G));//表示给这个数组初始化为十六进制最大值,大约1e9真好可以防止爆int。

AC代码如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define MAXN 1000
#define INF 0x3f3f3f3f
using namespace std;
int G[MAXN][MAXN];
int M[MAXN];
int answer[MAXN];
int main()
{
    memset(G,0x3f,sizeof(G));
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        int num,left ,r;
        scanf("%d%d%d",&num,&left,&r);
        M[i]=num;
        if(left==0)
        {
            continue;
        }
        else 
        {
            G[i][left]=1;
            G[left][i]=1;
        }
        if(r==0)
        {
          continue;
        }
        else 
        {
            G[i][r]=1;
            G[r][i]=1;
        }
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            if(i==j)
            {
                G[i][j]=0;
            }
        }
    }
    for(int k=1;k<=n;k++)
    {
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                if(G[i][j]>G[i][k]+G[k][j])
                {
                    G[i][j]=G[i][k]+G[k][j];
                }
            }
        }
    }
    int mi=INF;
    for(int k=1;k<=n;k++)
    {
        for(int i=1;i<=n;i++)
        {
            answer[k]=G[k][i]*M[i]+answer[k];
        }
        mi=min(mi,answer[k]);
    }
    printf("%d",mi);
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/LITTLESUNwl/p/10714749.html
今日推荐