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

题目

六一儿童节,老师带了很多好吃的巧克力到幼儿园。每块巧克力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;
    }
}
©️2020 CSDN 皮肤主题: 技术黑板 设计师:CSDN官方博客 返回首页