题解 | 匈牙利算法#素数伴侣#

素数伴侣

http://www.nowcoder.com/practice/b9eae162e02f4f928eac37d7699b352e

此题为匈牙利算法解决二分图最大匹配问题。我们可以把数据分为偶数,奇数两部分,然后进行配对(因为素数为一奇一偶的和)。

import math
def isPrime(x):
    if x<=3:
        return x>1
    for i in range(2, int(math.sqrt(x)+1)):
        if x%i ==0:
            return False
    else:
        return True

//该odd是否有匹配的even,返回boolean值   
def match(odd):
// 对偶数进行遍历,看是否与该奇数配对
    for i in range(n):
    // 若两数和为素数且该偶数在这轮循环中没有被访问过
        if isPrime(odd+ evens[i]) and vis[i]==0:
            // 将该偶数标记为已访问
        	vis[i]=1
			// p[i]=odd 为该偶数对应的odd储存值,如果该偶数没有奇数odd匹配,则为0,否则返回所匹配的odd
			// 如果p[i]还未匹配或者 p[i]已经有其他匹配的odd2,且该match(odd2)还有其他可以匹配的偶数(True)
            if p[i]==0 or match(p[i]):
            	// 取代原来的p[i]=odd2,变为 p[i]=odd
                p[i]=odd
                return True
    return False
                
    

while True:
    try:
        n = int(input())
        nums = list(map(int,input().split()))
        odds = []
        evens = []
        for num in nums:
            if num%2==0:
                evens.append(num)
            else:
                odds.append(num)
        m = len(odds)
        n = len(evens)
        s = 0
        p=[0]*n
        for num in odds:
        //每一次循环的vis必须要清空
            vis = [0]*n
            if match(num):
                s+=1
        print(s)
        
    except:
        break

全部评论
import math def isPrime(x): if x <= 1: return False for i in range(2, int(math.sqrt(x)+1)): if x%i ==0: return False else: return True
1 回复 分享
发布于 2022-05-12 00:07
当在函数f(b)中遇到被a占用的值,调用match(a)的时候,如果遇到了其他的匹配,(比如a也可以和4匹配,是应该返回True的) 但是p[i]已经在之前被自己占有了,不是又要再次调用match(a) 说得有点含糊,企图再次表达
点赞 回复 分享
发布于 05-08 15:12 贵州
有个地方不太理解,如果在match(1)中调用match去判断值是否有其他匹配值的时候,在循环里面即使遇到可以匹配的其他点;但是if p[i]==0 or match(p[i]):中因为p[i]已经被占用了,即使是该值就是自己,那不是又要调用一遍match(),没法进入阿(是不是要加上p[i]==j(也就是已经与该值匹配的对象就是自己) return True????不是很理解这个循环
点赞 回复 分享
发布于 05-08 14:39 贵州
31、41行有两个n,
点赞 回复 分享
发布于 2024-03-13 10:59 江苏
6
点赞 回复 分享
发布于 2023-01-17 18:43 广东
匈牙利算法
点赞 回复 分享
发布于 2022-06-19 21:48
才发现数据范围大于1,我傻了
点赞 回复 分享
发布于 2022-03-24 10:23

相关推荐

评论
32
13
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务