LeetCode-1615. 最大网络秩

1615. 最大网络秩

题目说明

n 座城市和一些连接这些城市的道路 roads 共同组成一个基础设施网络。每个 roads[i] = [ai, bi] 都表示在城市 ai 和 bi 之间有一条双向道路。

两座不同城市构成的 城市对 的 网络秩 定义为：与这两座城市 直接 相连的道路总数。如果存在一条道路直接连接这两座城市，则这条道路只计算 一次 。

整个基础设施网络的 最大网络秩 是所有不同城市对中的 最大网络秩 。

给你整数 n 和数组 roads，返回整个基础设施网络的 最大网络秩 。

示例

输入：n = 4, roads = [[0,1],[0,3],[1,2],[1,3]]
输出：4
解释：城市 0 和 1 的网络秩是 4，因为共有 4 条道路与城市 0 或 1 相连。位于 0 和 1 之间的道路只计算一次。

输入：n = 5, roads = [[0,1],[0,3],[1,2],[1,3],[2,3],[2,4]]
输出：5
解释：共有 5 条道路与城市 1 或 2 相连。

代码实现

class Solution:
    def maximalNetworkRank(self, n: int, roads: List[List[int]]) -> int:
        g = [[] for _ in range(n)]
        # 将城市对的信息存入数组
        for i, j in roads:
            g[i].append(j)
            g[j].append(i)
        ans = 0
        # 遍历数组，如果两个点不相连，网络秩=各个点相连的点的个数相加，如果相连，要-1
        for i in range(n):
            for j in range(i + 1, n):
                tmp = 1 if i in g[j] else 0
                ans = max(ans, len(g[i]) + len(g[j]) - tmp)
        return ans