首页 > 代码库 > nyoj 239 月老的难题 【二分匹配之匈牙利】
nyoj 239 月老的难题 【二分匹配之匈牙利】
月老的难题
时间限制:1000 ms | 内存限制:65535 KB
难度:4
- 描述
月老准备给n个女孩与n个男孩牵红线,成就一对对美好的姻缘。
现在,由于一些原因,部分男孩与女孩可能结成幸福的一家,部分可能不会结成幸福的家庭。
现在已知哪些男孩与哪些女孩如果结婚的话,可以结成幸福的家庭,月老准备促成尽可能多的幸福家庭,请你帮他找出最多可能促成的幸福家庭数量吧。
假设男孩们分别编号为1~n,女孩们也分别编号为1~n。
- 输入
- 第一行是一个整数T,表示测试数据的组数(1<=T<=400)
每组测试数据的第一行有两个整数n,K,其中男孩的人数与女孩的人数都是n。(n<=500,K<=10 000)
随后的K行,每行有两个整数i,j表示第i个男孩与第j个女孩有可能结成幸福的家庭。(1<=i,j<=n) - 输出
- 对每组测试数据,输出最多可能促成的幸福家庭数量
- 样例输入
1 3 4 1 1 1 3 2 2 3 2
- 样例输出
2
这道题, 用邻接矩阵TL, 得用邻接表
代码(链式前向星):
#include <stdio.h> #include <string.h> #include <algorithm> #define M 20005 using namespace std; struct node{ int to, next; }s[M]; int head[M], res[555], n; bool vis[555]; int find(int u){ int i; for(i = head[u]; i != -1; i = s[i].next){ int temp = s[i].to; if(!vis[temp]){ vis[temp] = 1; if(res[temp]== 0||find(res[temp])){ res[temp] = u; return true; } } } return false; } int main(){ //freopen("stdin.txt", "r", stdin); int t, k; scanf("%d", &t); while(t --){ scanf("%d%d", &n, &k); memset(res, 0, sizeof(res)); memset(head, -1, sizeof(head)); int a, b; for(int i = 0; i < k; i ++){ scanf("%d%d", &a,&b); s[i].to = b; s[i].next = head[a]; head[a] = i; } int ans = 0; for(int i = 1; i <= n; i ++){ memset(vis, 0, sizeof(vis)); if(find(i)) ++ans; } printf("%d\n", ans); } }
用vector内存有点大
#include <stdio.h> #include <string.h> #include <vector> #define M 555 using namespace std; vector <int > map[M]; int vis[M], res[M]; int n; int find(int a){ int i; for(i = 0; i < map[a].size(); i ++){ if(!vis[map[a][i]]){ vis[map[a][i]] = 1; if(res[map[a][i]] == 0||find(res[map[a][i]])){ res[map[a][i]] = a; return 1; } } } return 0; } int main(){ int t, k; scanf("%d", &t); while(t --){ scanf("%d%d", &n, &k); int i; memset(map, 0, sizeof(map)); memset(res, 0, sizeof(res)); int a, b, ans = 0; for(i = 0; i < k; i ++){ scanf("%d%d", &a, &b); map[a].push_back(b); } for(i = 1; i <= n; i ++){ memset(vis, 0,sizeof(vis)); if(find(i)) ++ans; } printf("%d\n", ans); } return 0; }
nyoj 239 月老的难题 【二分匹配之匈牙利】
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。