學習華為上機測試題，遇見了下面題，很有意思，核心是匈牙利算法問題。

特此學習記錄。資料均參考自網絡。

匈牙利算法目的：找出兩邊最大的匹配的數量。

參考資料：

https://blog.csdn.net/u013377068/article/details/79893013

https://blog.csdn.net/sunny_hun/article/details/80627351

https://www.nowcoder.com/profile/8408989/codeBookDetail?submissionId=25842225

題目描述：
若兩個正整數的和為素數，則這兩個正整數稱之為“素數伴侶”，如2和5、6和13，它們能應用于通信加密。現在密碼學會請你設計一個程序，從已有的N（N為偶數）個正整數中挑選出若干對組成“素數伴侶”，挑選方案多種多樣，例如有4個正整數：2，5，6，13，如果將5和6分為一組中只能得到一組“素數伴侶”，而將2和5、6和13編組將得到兩組“素數伴侶”，能組成“素數伴侶”最多的方案稱為“最佳方案”，當然密碼學會希望你尋找出“最佳方案”。

輸入:有一個正偶數N（N≤100），表示待挑選的自然數的個數。后面給出具體的數字，范圍為[2,30000]。

輸出:輸出一個整數K，表示你求得的“最佳方案”組成“素數伴侶”的對數。

輸入說明：1?輸入一個正偶數n， 2輸入n個整數

輸出描述: ?求得的“最佳方案”組成“素數伴侶”的對數。

# In[]# 匈牙利算法 # 別人代碼 def prime_judge(n): #判斷一個數是否是素數m=int(n**0.5)if n%2==0:return Falseelse:for i in range(m+1)[3::2]:if n%i==0:return Falsereturn Truedef group_lst(lst): #判斷列表內數為奇偶數，并分開存放a = []b = []for i in lst:if int(i)%2 == 1:a.append(int(i))else:b.append(int(i))return (a, b)def matrix_ab(a, b): #構建一個a行b列的二維矩陣，矩陣內容為1表示a+b為素數，為0表示a+b不是素數matrix = [[0 for i in range(len(b))] for i in range(len(a))]for ii, i in enumerate(a):for jj, j in enumerate(b):if prime_judge(i+j) == True:matrix[ii][jj] = 1return matrixdef find(x): #匈牙利算法匹配for index, i in enumerate(b):if matrix[x][index] == 1 and used[index] == 0: # 男女有好感 且 這個女的還沒有和男的匹配過used[index] = 1if connect[index] == -1 or find(connect[index]) != 0: # 如果這個女的還單身 或者 這個女的已經配好的男的還能去找別的女的（遞歸）connect[index] = x # 匹配成功 第x號男的 和 第index號女的 return 1return 0while True: #理解a,b為男女配對的話try:n = int(input())m = input().split()(a, b) = group_lst(m)matrix = matrix_ab(a, b)connect = [-1 for i in range(len(b))] #標記這個女的是否已經配對成功count = 0 for i in range(len(a)): used = [0 for j in range(len(b))] # 女的是否被這個男的查找過了if find(i): #從當前男的開始查找count += 1 # 配對成功+1（這個男女的配對可能會變，但是數量不變）print(count)except:break

?

?

?

?

?

總結

以上是生活随笔為你收集整理的匈牙利算法学习笔记_Python代码的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。