leetcode-1334

题意分析

1334. 阈值距离内邻居最少的城市 - 力扣（LeetCode）

        就是求给定一点，到其他点最短距离小于某个给定值，它的邻居数量。输出邻居数量最少的点。

算法思路

        看到任意2点的最短路，很直接可以想到floyd算法。floyd算法不难理解而且实现非常简单（下图），就是不断枚举插入新节点，看是不是可以使两点间路径更短。这可以算是经典题了。

        vector<vector<int>> dist(n,vector<int>(n,inf));
        for(int k=0;k<n;k++){
            for(int u=0;u<n;u++){
                for(int v=0;v<n;v++){
                    dist[u][v]=min(dist[u][v],dist[u][k]+dist[k][v]);
                }
            }
        }

代码实现

class Solution {
public:
    int findTheCity(int n, vector<vector<int>>& edges, int distanceThreshold) {
        const int inf=INT_MAX/2;
        vector<vector<int>> dist(n,vector<int>(n,inf));
        for(int k=0;k<n;k++){
            for(int u=0;u<n;u++){
                for(int v=0;v<n;v++){
                    dist[u][v]=min(dist[u][v],dist[u][k]+dist[k][v]);
                }
            }
        }
        int ans=-1;
        int min_nb=inf;
        for(int u=0;u<n;u++){//正序遍历求最大编号
            int nb=0;
            for(int v=0;v<n;v++){
                if(u!=v && dist[u][v]<inf){
                    nb++;
                }
            }
            if(min_nb>=nb){//要取到等
                min_nb=nb;
                ans=u;
            }
        }
        return ans;
    }
};

解题总结

        看到任意2点最短路径考虑用floyd算法。