华为OD 2024笔试机试 - 城市聚集度/找城市 (python C卷D卷真题算法)

华为OD机试（C卷+D卷）2024真题目录(Java & c++ & python)

题目描述

一张地图上有n个城市，城市和城市之间有且只有一条道路相连：要么直接相连，要么通过其它城市中转相连（可中转一次或多次）。城市与城市之间的道路都不会成环。

当切断通往某个城市 i 的所有道路后，地图上将分为多个连通的城市群，设该城市i的聚集度为DPi（Degree of Polymerization），DPi = max（城市群1的城市个数，城市群2的城市个数，…城市群m 的城市个数）。

请找出地图上DP值最小的城市（即找到城市j，使得DPj = min(DP1,DP2 … DPn))

提示：如果有多个城市都满足条件，这些城市都要找出来（可能存在多个解）

提示：DPi的计算，可以理解为已知一棵树，删除某个节点后；生成的多个子树，求解多个子数节点数的问题。

输入描述

每个样例：第一行有一个整数N，表示有N个节点。1 <= N <= 1000。

接下来的N-1行每行有两个整数x，y，表示城市x与城市y连接。1 <= x, y <= N

输出描述

输出城市的编号。如果有多个，按照编号升序输出。

用例1

输入

5
1 2
2 3
3 4
4 5

输出

3

说明
输入表示的是如下地图：

对于城市3，切断通往3的所有道路后，形成2个城市群[（1，2），（4，5）]，其聚集度分别都是2。DP3 = 2。

对于城市4，切断通往城市4的所有道路后，形成2个城市群[（1，2，3），（5）]，DP4 = max（3，1）= 3。

依次类推，切断其它城市的所有道路后，得到的DP都会大于2，因为城市3就是满足条件的城市，输出是3。

用例2

输入

6
1 2
2 3
2 4
3 5
3 6

输出

2 3

说明 将通往2或者3的所有路径切断，最大城市群数量是3，其他任意城市切断后，最大城市群数量都比3大，所以输出2 3

解题思路

用图解来解释一下用例1：

要求找出地图上DP值最小的城市（即找到城市j，使得DPj = min(DP1,DP2 … DPn))，

因此DP3最小，输出DP3的3。

实际就是求解删除一个节点后求最大的联通块大小=DPi，再取其中最小的DPi。

求联通块的方法有很多，可以建图后进行BFS或者DFS来得到联通块的大小，由于题集里上一题是用的并查集，这里偷懒直接复用一些前面代码，关于并查集的知识点可以到题集找上一题的图解学习。

删除某个节点=并查集合并时，跟这个节点相关的都跳过即可。

参考代码

import sys

# 输入获取
n = int(input())
relations = [list(map(int, input().split())) for i in range(n - 1)]

# 并查集
class UnionFindSet:
    def __init__(self, n):
        self.fa = [i for i in range(n)]

    def find(self, x):
        if x != self.fa[x]:
            self.fa[x] = self.find(self.fa[x])
        return self.fa[x]

    def union(self, x, y):
        x_fa = self.find(x)
        y_fa = self.find(y)
        if x_fa != y_fa:
            self.fa[y_fa] = x_fa

# 最小城市聚集度
minDp = sys.maxsize
# 城市列表
city = []

# 遍历每个城市 1~n
for i in range(1, n + 1):
    # 利用并查集对城市进行关联
    ufs = UnionFindSet(n + 1)

    for x, y in relations:
        # 切断城市的所有道路，即忽略和城市i有联系的合并操作
        if x == i or y == i:
            continue
        # 否则连接x和y
        ufs.union(x, y)

    # 统计各个连通分量自身的城市个数
    cnts = [0] * (n + 1)
    for j in range(1, n + 1):
        fa = ufs.find(j)
        cnts[fa] += 1

    # 取最多城市个数作为当前的切断城市的聚集度
    dp = max(cnts)

    if dp < minDp:
        minDp = dp
        city = [i]
    elif dp == minDp:
        city.append(i)

# 如果有多个，按照编号升序输出
city.sort()
print(" ".join(map(str, city)))