华为机试：HJ28 素数伴侣 java

描述

题目描述
若两个正整数的和为素数，则这两个正整数称之为“素数伴侣”，如2和5、6和13，它们能应用于通信加密。现在密码学会请你设计一个程序，从已有的 N （ N 为偶数）个正整数中挑选出若干对组成“素数伴侣”，挑选方案多种多样，例如有4个正整数：2，5，6，13，如果将5和6分为一组中只能得到一组“素数伴侣”，而将2和5、6和13编组将得到两组“素数伴侣”，能组成“素数伴侣”最多的方案称为“最佳方案”，当然密码学会希望你寻找出“最佳方案”。

输入:

有一个正偶数 n ，表示待挑选的自然数的个数。后面给出 n 个具体的数字。

输出:

输出一个整数 K ，表示你求得的“最佳方案”组成“素数伴侣”的对数。

数据范围： 1 \le n \le 100 \1≤n≤100  ，输入的数据大小满足 2 \le val \le 30000 \2≤val≤30000 

输入描述：

输入说明
1 输入一个正偶数 n
2 输入 n 个整数

输出描述：

求得的“最佳方案”组成“素数伴侣”的对数。

示例1

输入：

4
2 5 6 13

输出：

2

示例2

输入：

2
3 6

输出：

0

解题思路（匈牙利算法）

• 首先定义两个list容器，分别存储输入整数中的奇数和偶数。
• 然后利用匈牙利算法找到“素数伴侣”对数最多时的配对数。匈牙利算法的核心思想是先到先得，能让就让。
• 最后输出“素数伴侣”最多时的对数。

举例说明：首先A1和B2配对（先到先得），然后轮到A2，A2也可以和B2配对，这时候B2发现A1还可以和B4配对，所以放弃了A1，选择和A2组成伴侣（能让就让）。接着A3直接和B1配对（先到先得）。最后A4尝试与B4配对，但是这样A1就只能与B2配对，而A2就找不到伴侣了，一层层递归下来，发现不可行，所以A4不能与B4配对。

通过全部用例

运行时间31ms

占用内存10116KB

import java.io.*;
import java.util.*;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String line = br.readLine();
        int n = Integer.parseInt(line);
        //验证输入的n是否为1-100之间的正整数
        if (n % 2 != 0 || n < 1 || n > 100) {
            return;
        }
        line = br.readLine();
        //去除所有的空格
        String[] strs = line.split(" ");
        //验证第二次输入的数字是否符合要求
        for (String str : strs) {
            if (!str.matches("[0-9]+") || strs == null || strs.length == 0 || strs.length != n) {
                return;
            }
        }
        //用于存储所有的奇数
        ArrayList<Integer> odds = new ArrayList<>();
        //用于存储所有的偶数
        ArrayList<Integer> evens = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            int num = Integer.parseInt(strs[i]);
            if (num < 2 || num > 30000) {
                return;
            }
            //将奇数添加到odds
            if (num % 2 == 1) {
                odds.add(num);
            }
            //将偶数添加到evens
            if (num % 2 == 0) {
                evens.add(num);
            }
        }
        //下标对应已经匹配的偶数的下标，值(奇数)对应这个偶数的奇数伴侣,默认为0,表示没有奇数伴侣
        int[] matchEven = new int[evens.size()];
        //记录伴侣的对数
        int mates = 0;
        //遍历每一个奇数,代入find()方法
        for (int odd:odds) {
            //用于标记对应的偶数是否已经匹配,默认为false,未匹配
            boolean[] isUsed = new boolean[evens.size()];
            //如果匹配上，则计数加1
            if (find(odd, matchEven, evens, isUsed)) {
                mates++;
            }
        }
        System.out.println(mates);
    }
    
    //判断奇数odd能否找到伴侣
    private static boolean find(int odd, int[] matchEven, ArrayList<Integer> evens, boolean[] isUsed) {
        for (int i = 0; i < evens.size(); i++) {
            //该位置偶数没被访问过，并且和odd能组成素数伴侣
            if (isPrime(odd + evens.get(i)) && !isUsed[i]) {
                //改变该偶数的状态
                isUsed[i] = true;
                //如果i位置偶数还没有伴侣，则与odd组成伴侣，
                //如果已经有伴侣，并且这个(奇数)伴侣能重新找到(偶数)新伴侣，
                //则把原来伴侣让给别人，自己与odd组成伴侣
                if (matchEven[i] == 0 || find(matchEven[i], matchEven, evens, isUsed)) {
                    matchEven[i] = odd;
                    return true;
                }
            }
        }
        return false;
    }

    //对result进行判断,不是质数的的话返回false
    public static Boolean isPrime(int num) {
        if (num % 6 != 1 && num % 6 != 5) {
            return false;
        }
        for (int i = 2; i <= (int) Math.sqrt(num); i++) {
            if (num % i == 0) {
                return false;
            }
        }
        return true;
    }
}

看官觉得我写的还行的话，麻烦动动您可爱的小手，点个赞，

您的支持是我最大的动力！！！