当前位置: 代码迷 >> 综合 >> 六一儿童节(二分图最大匹配,Java实现)
  详细解决方案

六一儿童节(二分图最大匹配,Java实现)

热度:54   发布时间:2023-11-22 00:04:01.0

题目

六一儿童节,老师带了很多好吃的巧克力到幼儿园。每块巧克力j的重量为w[j],对于每个小朋友i,当他分到的巧克力大小达到h[i] (即w[j]>=h[i]),他才会上去表演节目。老师的目标是将巧克力分发给孩子们,使得最多的小孩上台表演。可以保证每个w[i]> 0且不能将多块巧克力分给一个孩子或将一块分给多个孩子。

输入描述:
第一行:n,表示h数组元素个数
第二行:n个h数组元素
第三行:m,表示w数组元素个数
第四行:m个w数组元素

输出描述:
上台表演学生人数

示例1
输入
3
2 2 3
2
3 1
输出
1

题解

乍一看有点像贪心,但是看到一个人对应一个巧克力时,就很明显就是二分图最大匹配了。

代码

import java.io.*;
import java.math.*;
import java.util.*;public class Main{public static int n,m;public static int []h=new int[1007];public static int []w=new int[1007];public static int [][]g=new int[1007][1007];public static void main(String[] args){Scanner in=new Scanner(System.in);n=in.nextInt();for(int i=0;i<n;i++){h[i]=in.nextInt();}m=in.nextInt();for(int i=0;i<m;i++){w[i]=in.nextInt();}for(int i=0;i<n;i++){for(int j=0;j<m;j++){if(h[i]<=w[j]){g[i][j]=1;}else {g[i][j]=0;}}}int ans=solve();System.out.println(ans);}public static int[] pre=new int[1007];public static int[] vis=new int[1007];public static int Find(int u){for(int v=0;v<m;v++){if(g[u][v]==0) continue;if(vis[v]==1) continue;vis[v]=1;if(pre[v]==-1 || Find(pre[v])==1){pre[v]=u;return 1;}}return 0;}public static int solve(){Arrays.fill(pre,-1);int ans=0;for(int i=0;i<n;i++){Arrays.fill(vis,0);if(Find(i)==1){ans++;}}return ans;}
}
  相关解决方案