二分圖的最大匹配(匈牙利演算法)程式碼

longwind7發表於2024-05-19

二分圖的最大匹配

程式碼

#include <bits/stdc++.h>
using namespace std;

const int N = 505, M = 100005;
int h[N], e[M], ne[M], idx;
int match[N];
bool st[N];
int n1, n2, m;

void add(int a, int b)
{
    e[idx] = b; //e[idx]存放的是第idx條邊的終點
    ne[idx] = h[a]; //ne[idx]存放的是與a結點相連的下一條邊
    h[a] = idx++; //把當前這條邊放入
}

bool find(int x)
{
    for (int i = h[x]; i != -1; i = ne[i])
    {
        int j = e[i];
        if (!st[j])
        {
            st[j] = true;
            if (match[j] == 0 || find(match[j]))
            {
                match[j] = x;
                return true;
            }
        }
    }
    return false;
}

int main()
{
    cin >> n1 >> n2 >> m;

    memset(h, -1, sizeof h); // 將所有結點連線的邊初始化為-1

    while (m--)
    {
        int a, b;
        cin >> a >> b;
        add(a, b); //將每個邊加入到鄰接表中
    }

    int ans = 0;
    for (int i = 1; i <= n1; i++)
    {
        memset(st, false, sizeof st); //每次都要初始化每個女孩的狀態,避免遞迴出現死迴圈
        if (find(i)) //如果當前左節點匹配右節點成功
            ans++;
    }

    cout << ans << endl;
    return 0;
}

相關文章