[python刷题模板] 匈牙利算法

一、 算法&数据结构

1. 描述

匈牙利算法用来计算二分图的最大匹配问题。

2. 复杂度分析

1. nmm

3. 常见应用

1. 最大配对

4. 常用优化

1. used每次都要置false，可以用颜色来区分，不是当前颜色就是false，就不用每次都n方做一遍了。

二、 模板代码

1. 二分图的最大匹配，标准匈牙利

例题: P3386 【模板】二分图最大匹配

import sys
from collections import *
from itertools import *
from math import *
from array import *
from functools import lru_cache
import heapq
import bisect
import random
import io, os
from bisect import *

if sys.hexversion == 50924784:
    sys.stdin = open('cfinput.txt')

RI = lambda: map(int, sys.stdin.buffer.readline().split())
RS = lambda: map(bytes.decode, sys.stdin.buffer.readline().strip().split())
RILST = lambda: list(RI())

MOD = 10 ** 9 + 7
"""
"""


#  	 ms
def solve(n, m, e, es):
    g = [[] for _ in range(n)]
    for u, v in es:
        g[u - 1].append(v - 1)

    ans = [-1] * m
    vis = [-1] * m

    def find(i, c):
        for j in g[i]:
            if vis[j] != c:
                vis[j] = c
                if ans[j] == -1 or find(ans[j], c):
                    ans[j] = i
                    return True
        return False

    cnt = 0
    for i in range(n):
        # vis = [-1]*m
        if find(i, i):
            cnt += 1
    print(cnt)


if __name__ == '__main__':
    n, m, e = RI()
    es = []
    for _ in range(e):
        es.append(RILST())
    solve(n, m, e, es)

三、其他

四、更多例题

五、参考链接