leetcode 787 k次中转的单源最短路径*

最新推荐文章于 2022-04-16 12:05:45 发布

qq_45992231

最新推荐文章于 2022-04-16 12:05:45 发布

阅读量146

点赞数

分类专栏： leetcode 文章标签：算法

本文链接：https://blog.csdn.net/qq_45992231/article/details/119887350

版权

leetcode 专栏收录该内容

53 篇文章 0 订阅

订阅专栏

动态规划

#define MAXN 0x7ffffff
#define MIN(x,y) (x<y?x:y)
int findCheapestPrice(int n, int** flights, int flightsSize, int* flightsColSize, int src, int dst, int k){
    int martrix[n][n];
    int dis0[n];//存储src到各个点经过k-1次中转的距离
    int dis1[n];//存储src到各个点经过k次中转的距离
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            martrix[i][j]=MAXN;
        }
    }
    for(int i=0;i<flightsSize;i++)
    {
        martrix[flights[i][0]][flights[i][1]]=flights[i][2];
    }
    //初始化dis0数组
    for(int i=0;i<n;i++)
    {
        dis0[i]=martrix[src][i];
    }
    int minn=MAXN;
    for(int t=0;t<=k;t++)
    {
        //考察t中转次数时src到dst的最短距离， 初始时中转次数是0
        if(dis0[dst]!=MAXN)
           minn=MIN(minn,dis0[dst]);
        if(t==k)
           break;
        //通过dis0数组，生成dis1数组，即中转次数+1时的距离
        for(int i=0;i<n;i++)
        {
            dis1[i]=MAXN;
        }
        for(int i=0;i<n;i++)
        {
            if(i!=src)
            {
                for(int j=0;j<n;j++)
                {
                    if(dis0[j]!=MAXN&&martrix[j][i]!=MAXN)
                    {
                        dis1[i]=MIN(dis1[i],dis0[j]+martrix[j][i]);
                    }
                }
            }
        }
        //改为dis0数组
        for(int i=0;i<n;i++)
        {
            dis0[i]=dis1[i];
        }
    }
    if(minn==MAXN)
    return -1;
    return minn;
}