匹配算法面试题

一、二分图匹配(匈牙利算法)解决红娘问题模板。
曾用此模板A过网易笔试题，自制模板。本质是个深搜。

# -*- coding: utf-8 -*-
m,n=map(int,input().split(" "))#m:左侧集合的数量 n:右侧集合的数量。m为男孩的数量，n为女孩的数量。
book=[[0]*n for i in range(m)]#邻接矩阵存图，M行N列.book[i][j]:代表红娘问题中的i男生喜欢j女生
for i in range(n):
    a,b=map(int,input().split(" "))
    book[a][b]=1
cnt=0#能匹配的最多对数
p=[0]*n#记录右侧元素对应的左侧元素
vis=[0]*n#记录右侧元素是否被访问过

def match(i):
    for j in range(n):
        if(book[i][j]==1 and vis[j]==0):#如果i,j可以连接,j没有被访问过。
            vis[j]=1    #记录状态为访问过
            if(p[j]==-1 or match(p[j])):#如果j暂无匹配或者原来匹配的左侧元素可以找到新的匹配元素
                p[j]=i#j匹配i
                return True
    return False

for i in range(m):
    vis=[0]*n
    #对于左侧集合中的每个元素
    if(match(i)):
        cnt+=1
print(cnt)

二、判断二分图(LeetCode 785)

class Solution:
    def isBipartite(self, graph: List[List[int]]) -> bool:
        def dfs(book,colors,i,color,N):
            colors[i]=color
            for j in range(N):
                if(book[i][j]==1):
                    if(colors[j]==color):
                        return False
                    if(colors[j]==0 and not dfs(book,colors,j,-1*color,N)):
                        return False
            return True

        N=len(graph)
        book=[[0]*N for i in range(N)]
        colors=[0]*N
        for i in range(N):
            for j in graph[i]:
                book[i][j]=1
        for i in range(N):
            if(colors[i]==0 and not dfs(book,colors,i,1,N)):
                return False
        return True