素数伴侣算法

//两数相加要为素数，则这两个数必然一个奇数一个为偶数
//将输入的数据分为奇数和偶数两部分


#include<string>
#include<iostream>
#include<vector>
using namespace std;
#define MAX_COUNT 100
vector<int> even;                //存储偶数
vector<int> odd;    
vector<vector<int>> line(MAX_COUNT);//存储奇数
//int line[MAX_COUNT][MAX_COUNT];  //标记以此下标(x,y)的两个数之和是素数
vector<int> used;             //标记此下标对应的奇数值是否被使用
int evenCount = 0;               //偶数计数
int oddCount = 0;                 //奇数计数
/* 判断是不是素数 */
int IsPrime(int num)
{
    for (int i = 2; i * i <= num; i++) {
        if (num % i == 0) {
            return 0;
        }
    }
    return 1;
}


/* 将数组的元素分为偶数和奇数两部分，并做出标记 */
void IsEvenOrOdd(vector<int> num)
{
    for (auto x : num) {
        if (x % 2 == 0) {
            even.push_back(x);
            evenCount++;
        }
        else {
            odd.push_back(x);
            oddCount++;
        }
    }
}
vector<int> oddEven(MAX_COUNT,-1); //存储第j个奇数匹配的偶数下标
/* 匈牙利算法，直到匹配的数，返回1，否则返回0 */

/*
int line[MAX_COUNT][MAX_COUNT];  //标记以此下标(x,y)的两个数之和是素数
int used[MAX_COUNT];             //标记此下标对应的奇数值是否被使用
int oddEven[MAX_COUNT]; //存储第j个奇数匹配的偶数下标
*/


int func(int n) {
    for (int i = 0; i < oddCount; i++) {
        if (IsPrime(even[n] + odd[i]) == 1 && used[i] == 0){
            used[i] = 1;
            if (oddEven[i] == -1 || func(oddEven[i])){
                oddEven[i]=n;
                return 1;
            }
        }
    }
    return 0;
}

int main()
{
    int n = 0;
    int sum = 0;
    vector<int> num;
    cin >> n;

    for (int i = 0; i < n; i++) {
        int temp;
        cin >> temp;
        num.emplace_back(temp);
    }
    IsEvenOrOdd(num);

    /* 遍历所有偶数，查找出与之匹配的奇数 */
    for (int i = 0; i < evenCount; i++) {
        /* 每一轮清除used标记 */
        used.clear();
        for (int i = 0; i < MAX_COUNT; i++) {
            used.emplace_back(0);
        }
        sum += func(i);
    }
    printf("%d", sum);
    return 0;
}