匈牙利算法-素數伴侶

代碼基本上參考了博客[1],內含原理闡釋。我實現的判斷素數算法,減少for循環(huán)運行的次數,運行速度較快。

#include <iostream>
#include <math.h>
#include <vector>
#include <memory.h>
using namespace std;
static inline bool isPrime(int num){
    bool ret=false;
    if(num<=1||(num>2&&0==(num%2))){
        return ret;
    }
    if(2==num||3==num||5==num||7==num){
        ret=true;
        return ret;
    }
    if(0==(num%3)||0==(num%5)||0==(num%7)){
        return ret;
    }
    int limit=sqrt(num);
    ret=true;
    for(int i=11;i<=limit;i+=2){
        if(0==(num%i)){
            ret=false;
            break;
        }
    }
    return ret;
}
class Solution{
public:
    int maxPair(const std::vector<int>&data){
        int sz=data.size();
        std::vector<int> odds,evens;
        int res=0;
        for(int i=0;i<sz;i++){
            if(0==(data.at(i)%2)){
                evens.push_back(data.at(i));
            }else{
                odds.push_back(data.at(i));
            }
        }
        int m=evens.size(),n=odds.size();
        if(0==m||0==n){
            return res;
        }
        uint8_t used[m];
        std::vector<int> even_matchs;
        even_matchs.resize(m,0);
        for(int i=0;i<n;i++){
            memset(used,0,m*sizeof(uint8_t));
            if(find(odds.at(i),evens,even_matchs,used)){
                res++;
            }
        }
        return res;
    }
private:
    bool find(int odd,std::vector<int>&evens,
              std::vector<int>&even_matchs,uint8_t *used){
        for(int i=0;i<evens.size();i++){
            if((0==used[i])&&isPrime(odd+evens.at(i))){
                used[i]=1;
                if(0==even_matchs[i]||
                   find(even_matchs[i],evens,even_matchs,used)){
                    even_matchs[i]=odd;
                    return true;
                }
            }
        }
        return false;
    }
};
int main()
{
    std::vector<int> data;
    //{20923,22855 ,2817 ,1447 ,29277, 19736 ,20227 ,22422, 24712 ,27054, 27050, 18272, 5477, 27174,
     //13880, 15730 ,7982, 11464 ,27483, 19563, 5998, 16163};
    int n;
    std::cin>>n;
    for(int i=0;i<n;i++){
        int num;
        std::cin>>num;
        data.push_back(num);
    }
    Solution so;
    int ret=so.maxPair(data);
    std::cout<<ret<<std::endl;
    return 0;
}

Reference:
[1]二分圖最大匹配(匈牙利算法):素數伴侶https://www.cnblogs.com/xiehuazhen/p/12574797.html

?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
【社區(qū)內容提示】社區(qū)部分內容疑似由AI輔助生成,瀏覽時請結合常識與多方信息審慎甄別。
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發(fā)布,文章內容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內容

友情鏈接更多精彩內容