HDU1845 - Jimmy’s Assignment

最大匹配  Hopcroft - Karp 算法

#include <cstring>
#include <iostream>
#include <cstdlib>
#include <cctype>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
#include <queue>
#define INF (int)(1e9)
#define maxn 5001
using namespace std;
typedef long long ll;
int p = 0;
struct node {
    int to, w, next;
};
node edge[maxn << 2];
int head[maxn];
void add_edge(int u, int v, int w) {
    edge[p].to = v;
    edge[p].w = w;
    edge[p].next = head[u];
    head[u] = p;
    ++ p;
}
int Mx[maxn],My[maxn],uN,vN;
int dx[maxn],dy[maxn],dis;
bool vis[maxn];
bool search_path() {
    queue<int> Q;
    dis = INF;
    memset(dx, -1, sizeof(dx));
    memset(dy, -1, sizeof(dy));
    for (int i = 0; i < uN; i++)
        if(Mx[i] == -1) {
            Q.push(i);
            dx[i] = 0;
        }
    while (!Q.empty()) {
        int u = Q.front();
        Q.pop();
        if (dx[u] > dis)  break;
        int v;
        for (int k = head[u]; k != -1; k = edge[k].next) {
            v = edge[k].to;
            if (dy[v] == -1) {
                dy[v] = dx[u] + 1;
                if (My[v] == -1)  dis = dy[v];
                else {
                    dx[My[v]] = dy[v]+1;
                    Q.push(My[v]);
                }
            }
        }
    }
    return dis != INF;
}
bool dfs(int u) {
    int v;
    for (int k = head[u]; k != -1; k = edge[k].next) {
        v = edge[k].to;
        if (!vis[v] && dy[v]==dx[u]+1) {
            vis[v] = 1;
            if (My[v] != -1 && dy[v] == dis) continue;
            if(My[v] == -1 || dfs(My[v])) {
                My[v] = u;
                Mx[u] = v;
                return true;
            }
        }
    }
    return false;
}
int max_match() {
    int res = 0;
    memset(Mx, -1, sizeof(Mx));
    memset(My, -1, sizeof(My));
    while (search_path()) {
        memset(vst, 0, sizeof(vst));
        for (int i = 0; i < uN; i ++)
            if (Mx[i] == -1 && dfs(i))  res ++;
    }
    return res;
}
int main() {
    int t;
    scanf("%d", &t);
    while (t --) {
        p = 0;
        memset(head, -1, sizeof(head));
        int n;
        scanf("%d", &n);
        for (int i = 0; i < (3*n/2); ++ i) {
            int u, v;
            scanf("%d%d", &u, &v);
            -- u, -- v;
            add_edge(u,v,0);
            add_edge(v,u,0);
        }
        uN = vN = n;
        cout << max_match()/2 << endl;
    }
}