匈牙利算法实现demo

下面是匈牙利算法的实现的简单例子：

数据：in.txt中的内容

1 1
2 2
2 3
3 1
3 2
3 3
3 4
4 2
4 3
5 2
5 3
5 5
5 6

代码：

#include <cstring>
#include <iostream>
using namespace std;

int map[10][10];
int pre[10];
bool flag[10];
int find(int cur) {
    for(int i=1; i<=6; ++i) {
        if(map[cur][i]&&!flag[i]) {
            flag[i]=true;
            if(pre[i]||find(pre[i])) {//如果未匹配或者匹配的其他点但还有增广路径。进行更新
                pre[i]=cur;
                return 1;
            }
        }
    }
    return 0;
}
int main() {
    freopen("in.txt","r",stdin);
    int a,b;
    while(scanf("%d%d",&a,&b)==2) {
        map[a][b]=1;
    }
    int ans=0;
    memset(pre,-1,sizeof(pre));
    for(int i=1; i<=5; ++i) {
        memset(flag,false,sizeof(flag));
        ans+=find(i);
    }
    cout<<ans<<endl;
    return 0;
}