4298 搭档(二分图的最大匹配、贪心)

1. 问题描述:

有 n 个男孩,每个人的魅力值依次为 a1,a2,…,an,有 m 个女孩,每个人的魅力值依次为 b1,b2,…,bm。学校举办舞会,希望选出尽可能多的男女搭档参加,已知当一对男女的魅力值相差不超过 1 时,他们才会愿意组成搭档。请问,最多可以同时凑出多少对搭档参加舞会。注意:不能男男配对或女女配对。每个男生最多只能和一个女生配对,反之亦然。

输入格式

第一行包含整数 n,第二行包含 n 个整数 a1,a2,…,an。第三行包含整数 m。第四行包含 m 个整数 b1,b2,…,bm。

输出格式

一个整数,表示最大可能对数

数据范围

前 6 个测试点满足 1 ≤ n,m ≤ 5;
所有测试点满足 1 ≤ n,m ≤ 100,1 ≤ ai,bi ≤ 100;

输入样例1:

4
1 4 6 2
5
5 1 5 7 9

输出样例1:

3

输入样例2:

4
1 2 3 4
4
10 11 12 13

输出样例2:

0

输入样例3:

5
1 1 1 1 1
3
1 2 3

输出样例3:

2
来源:https://www.acwing.com/problem/content/description/4301/

2. 思路分析:

分析题目可以知道属于男女配对的问题,一般男女配对,舞会配对的题目与二分图都有很大的关系,由题目可知我们需要求解二分图的最大匹配,一般求解二分图的最大匹配可以使用匈牙利算法,时间复杂度为 O(n ^ 3),但是一般达不到这个理论的时间复杂度,而题目数据范围在 100 以内所以肯定是可以通过的。我们可以根据题目的限制条件建立有向图,然后使用匈牙利算法的模板即可。除了使用匈牙利算法求解之外还可以使用贪心来求解,一般对于贪心的题目都有类似的套路,当题目中涉及到比较多方案的时候我们看能不能挖掘一下题目的性质,使得全集最优解可以在满足某种性质的小集合中求解得到,可以发现如果最优解中存在某两对交叉的情况我们可以通过交换对应的顺序,满足题目限制的条件下使得配对的时候不交叉而且总的配对情况不变,这样我们就可以发现小集合中存在一个全集最优解,那么小集合的最优解一定是全集最优解,这样我们在小集合中找到所有不交叉的配对情侣即可。我们可以将男孩和女孩的魅力值从小到大进行排序,从前往后开始枚举依次匹配即可,可以使用双指针来实现匹配的过程。

一般证明贪心思路是正确的可以证明两个方向成立,也即证明贪心解 >= 最优解和最优解 <= 贪心解,并且在证明的时候可以使用反证法来证明,反证法一般是假设逆否命题成立,然后推出矛盾那么就可以证明出原命题是成立的。

3. 代码如下:

匈牙利算法:

from typing import List


class Solution:
    # 判断当前的节点u是否可以匹配
    def dfs(self, u: int, match: List[int], st: List[int], g: List[List[int]]):
        for next in g[u]:
            if st[next] == -1:
                st[next] = 1
                if match[next] == -1 or self.dfs(match[next], match, st, g):
                    match[next] = u
                    return True
        return False

    def process(self):
        n = int(input())
        a = list(map(int, input().split()))
        m = int(input())
        b = list(map(int, input().split()))
        g = [list() for i in range(n + 10)]
        for i in range(1, n + 1):
            for j in range(1, m + 1):
                # 当魅力值相差不超过1的时候说明两个节点可以匹配
                if abs(a[i - 1] - b[j - 1]) <= 1:
                    # 建图
                    g[i].append(j)
        # match[i] = j表示右边的节点i有左边的节点j匹配
        match = [-1] * (m + 10)
        res = 0
        for i in range(1, n + 1):
            st = [-1] * (m + 10)
            if self.dfs(i, match, st, g):
                res += 1
        return res


if __name__ == '__main__':
    print(Solution().process())

贪心:

class Solution:
    def process(self):
        n = int(input())
        a = list(map(int, input().split()))
        m = int(input())
        b = list(map(int, input().split()))
        a.sort()
        b.sort()
        # 双指针实现
        i = j = res = 0
        while i < n and j < m:
            # 当前的男女可以配对
            if abs(a[i] - b[j]) <= 1:
                i += 1
                j += 1
                res += 1
            # 男的魅力值太小了那么需要往后移动
            elif a[i] < b[j]:
                i += 1
            else:
                j += 1
        return res


if __name__ == '__main__':
    print(Solution().process())
  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值