第十五届蓝桥杯大赛软件赛省赛Python 大学 B 组试做（下）【本期题单： 缴纳过路费， 纯职业小组】

本期题单：缴纳过路费，纯职业小队

缴纳过路费

题目

题目链接：缴纳过路费

思路分析

这一题就是很典型的 并查集 的题目了，它就是一个找父节点的思路流程。

对于并查集，本篇就不说了，后面会出一期来讲它的，如果对这个算法陌生的话可以学习一下这个大佬的文章：跳转链接，这位大佬讲的很好，我也是学习的他的文章。

下面就针对这道题分析一下 ——

题目里面比较需要注意的点 ——

1. 过路费是指最大价值，而不是总和
2. 过路费要处于上下限之间
3. 道路是双向的

规则一 和 规则二，就意味着如果有一条路的价值超过上限，那就不需要考虑它。对于低于下限的道路，可以和符合条件的道路共存，但不能单独存在。

规则三意味着没有直接连接的两个地方也能够到达。

OK，大概就是这些内容，然后我们针对下面这组数据来分析一下

6 8 2 7
1 2 2
1 3 5
2 3 8
1 4 1
2 4 5
2 5 4
3 4 9
2 6 1

其中 2 - 3 和 3 - 4这两条路超过了最大价值，可以在存储的时候直接去除掉，我们用不到它。
然后 1 - 4 和 2 - 6这两条路低于最低价值，我们需要记录一下，不能让它们单独出现。

uf1是低于上限的，uf2是低于下限的
他们的一个路径大致如下图 ——

也就是说——
（1，2）（1，3）（1，4）（1，5）（2，3）（2，4）（2，5）（3，4）（3，5）（4，5）
任选两个城市都是符合的，所以我们只需要知道符合的城市的总数，然后就能直接计算：

n * (n - 1) // 2

然后还需要注意类似uf2这样结构的，就是两个独立的分支，两边都需要计算。

低于上限的所有路径 - 只包含低于下限的路径

这样就能够计算上所有符合的了。
OKK，下面看看代码吧。

代码

class UnionFind:
    def __init__(self, n):
        self.pre = [i for i in range(n)]        # 记录父节点
        self.size = [1] * n
    def find(self, x):                          # 寻找根节点
        if x != self.pre[x]:                    # 判断父节点是不是自己本身
            rx = self.find(self.pre[x])         # 向上寻找父节点
            self.pre[x] = rx
            return rx
        return x
    def union(self, x, y):                      # 组合/连接两个节点
        rx, ry = self.find(x), self.find(y)
        if rx == ry:                            # 如果两个数的父节点是相同的就直接返回
            return
        self.pre[ry] = rx                   # 这里互换也是一样的，不考虑内部结构
        self.size[rx] += self.size[ry]      # 记录以它为根节点的所有子树


def cal(num):
    return num * (num - 1) // 2


n, m, l, r = map(int, input().split())
uf1 = UnionFind(n + 1)
uf2 = UnionFind(n + 1)
for i in range(n):
    u, v, w = map(int, input().split())
    if w <= r:                                  # 比上限小的才存进来
        uf1.union(u, v)
        if w < l:
            uf2.union(u, v)

ans = 0
for i in range(1, n + 1):
    if i == uf1.find(i):
        ans += cal(uf1.size[i])
    if i == uf2.find(i):
        ans -= cal(uf2.size[i])
print(ans)

纯职业小组

题目

题目链接：纯职业小组

思路分析

这道题就是去考虑最差情况，题目情况很多，我们一点点来分析。
首先我们对各个小组人数进行一个分类 ——

1. 比3小的，这个就直接加上去就行了
2. 比3大的，余数是 0 的，如果被用完 + 0，没用完 + 2
3. 比3大的，余数是 1 的，如果被用完 + 1，没用完 + 2
4. 比3大的，余数是 2 的，这个无论有没有被用完，都可以直接 + 2

给大家一个例子

1
2 2
1 8
3 4

你们按照上面的逻辑去写，就知道怎么个事儿了。

然后对于 所能组成的队伍数 - k（需要组成的队伍数） 如果 大于兵种数，那么就需要加上 2 * 兵种数，如果小于兵种数就需要进行更麻烦的计算了。

if k <= y2:
    ans += 3 * k + (y0 + y1) * 2 + y2 * 2
elif k <= y1 + y2:
    x = y1 + y2 - k
    ans += 3 * k + (y0 + x) * 2 + y2 * 2 + (y1 - x)
elif k <= y0 + y1 + y2:
    x = y0 + y1 + y2 - k
    ans = 3 * k + x * 2 + y2 * 2 + y1

大概就是这么回事。
然后和大伙说一下就是我的代码没有完全实现，至少在我对这道题目的理解下，但是能够解决题目测试点，但是整体思路应该是对的。

t = int(input())
for _ in range(t):
    n, k = map(int, input().split())
    A = {}
    z = 0
    for __ in range(n):
        a, b = map(int, input().split())
        if a in A.keys():
            A[a] += b
        else:
            A[a] = b
    group = ans = 0
    y0 = y1 = y2 = 0

    for i in A.keys():
        if A[i] >= 3:
            group += A[i] // 3
            if A[i] % 3 == 0:       # 余数为0
                y0 += 1
            elif A[i] % 3 == 1:     # 余数为1
                y1 += 1
            elif A[i] % 3 == 2:     # 余数为2
                y2 += 1
        else:
            ans += A[i]             # 比3小的队伍直接加
            z += 1
    if k > group:
        print(-1)
    else:
        if group - k > n:           # 测试样例全是这里的值
            ans += k * 3 + 2 * (len(A.keys()) - z)
            print(ans - 2)
        else:
            if k <= y2:
                ans += 3 * k + (y0 + y1) * 2 + y2 * 2
            elif k <= y1 + y2:
                x = y1 + y2 - k
                ans += 3 * k + (y0 + x) * 2 + y2 * 2 + (y1 - x)
            elif k <= y0 + y1 + y2:
                x = y0 + y1 + y2 - k
                ans = 3 * k + x * 2 + y2 * 2 + y1
        print(ans)

感谢大伙观看，别忘了三连支持一下

大家也可以关注一下我的其它专栏，同样精彩喔~

下期见咯~