牛客[HJ28 素数伴侣]

0. 题目

描述
题目描述
若两个正整数的和为素数，则这两个正整数称之为“素数伴侣”，如2和5、6和13，它们能应用于通信加密。现在密码学会请你设计一个程序，从已有的 N （ N 为偶数）个正整数中挑选出若干对组成“素数伴侣”，挑选方案多种多样，例如有4个正整数：2，5，6，13，如果将5和6分为一组中只能得到一组“素数伴侣”，而将2和5、6和13编组将得到两组“素数伴侣”，能组成“素数伴侣”最多的方案称为“最佳方案”，当然密码学会希望你寻找出“最佳方案”。

输入:

有一个正偶数 n ，表示待挑选的自然数的个数。后面给出 n 个具体的数字。

输出:

输出一个整数 K ，表示你求得的“最佳方案”组成“素数伴侣”的对数。

数据范围： 1 \le n \le 100 \1≤n≤100 ，输入的数据大小满足 2 \le val \le 30000 \2≤val≤30000
输入描述：
输入说明
1 输入一个正偶数 n
2 输入 n 个整数

输出描述：
求得的“最佳方案”组成“素数伴侣”的对数。

示例1
输入：
4
2 5 6 13
复制
输出：
2
复制
示例2
输入：
2
3 6
复制
输出：
0

1.思路

匈牙利算法：https://zhuanlan.zhihu.com/p/96229700
想得两个数配对得到素数，只有 奇数+偶数 才能得到素数，因此，我们将输入arr中的数字分成两个数组，分别存储奇数和偶数。之后，问题就转化为了二分图的匹配问题，利用匈牙利算法求解即可（匈牙利算法：先到先得，能让就让）

2. 代码

#判断是否为素数
def prime_judge(n):
    m = int(n ** 0.5) + 2
    for i in range(3, m, 2):
        if n % i == 0:
            return False
    return True
    
#将arr分为奇arr和偶arr
def group_lst(lst): 
    a, b = [], []
    for i in lst:
        if int(i) % 2 == 0:
            a.append(int(i))
        else:
            b.append(int(i))
    return a, b
    
#得到匹配矩阵，存储a与b是否能匹配   
def matrix_ab(a, b):
    l = [[0 for _ in b] for _ in a]
    for ii, i in enumerate(a):
        for jj, j in enumerate(b):
            if prime_judge(i + j):
                l[ii][jj] = 1
    return l
    
#匹配算法      
def find(x):
    for index in range(len(b)):#依次便利b中的元素
    	#如果b能与a匹配且b在该次匹配中未被使用，则标记b被使用
        if matrix[x][index] == 1 and used_b[index] == 0:
            used_b[index] = 1
            #递归，先到先得，能让就让
            #如果在该次匹配中，b还没有被配对则将b配对给a
            #若果在该次匹配中，b已经被陪陪给c，则判定c是否还有其他的匹配对象
            if conection_b[index] == -1 or find(conection_b[index]) != 0:
                conection_b[index] = x
                return 1
    return 0
      
while True:
    try:
        n = int(input())
        m = input().split(' ')
        a, b = group_lst(m)
        matrix = matrix_ab(a, b)
        conection_b = [-1 for _ in range(len(b))]
        count = 0
        for i in range(len(a)):
            used_b = [0 for _ in range(len(b))]
            if find(i):
                count += 1
        print(count)
    except:
        break