题解 | #素数伴侣#

素数伴侣

https://www.nowcoder.com/practice/b9eae162e02f4f928eac37d7699b352e

const rl = require("readline").createInterface({ input: process.stdin });
var iter = rl[Symbol.asyncIterator]();
const readline = async () => (await iter.next()).value;

void (async function () {
    // 输入数据
    const n = parseInt(await readline());
    const nums = (await readline()).split(" ").map(Number);
    const odds = nums.filter((item) => item % 2),
        evens = nums.filter((item) => item % 2 === 0);
    // 判断素数
    const isPrime = (num) => {
        const n = Math.floor(Math.sqrt(num))+1;
        for (let i = 2; i < n; i++) {
            if (num % i === 0) return false;
        }
        return num != 1;
    };
    // 匈牙利算法
    const find = (odd, evens, evenMatch, match) => {
        for (let i = 0; i < evens.length; i++) {
            const even = evens[i];
            // 如果这个偶数未被占用,并且和奇数的和是素数
            if (!match[i] && isPrime(even + odd)) {                
                match[i] = true;// 标记这个偶数已被占用
                // 如果这个偶数没有伴侣,或者它的伴侣可以换一个其他的偶数
                if (
                    evenMatch[i] === 0 ||
                    find(evenMatch[i], evens, evenMatch, match)
                ) {                    
                    evenMatch[i] = odd;// 更新这个偶数的伴侣为当前的奇数
                    return true;
                }
            }
        }
        return false;
    };

    //计算结果
    const calculate = () => {
        const evenMatch = new Array(evens.length).fill(0); //记录每个偶数的伴侣奇数
        let cnt = 0;
        // 对每个奇数进行匹配操作
        for (const odd of odds) {
            const match = new Array(evens.length).fill(false); // 创建一个数组,记录每个偶数是否已被占用
            if (find(odd, evens, evenMatch, match)) {
                cnt++; // 如果找到了素数伴侣,就增加计数
            }
        }
        return cnt;
    };
    console.log(calculate());
})();

全部评论

相关推荐

自从我室友在计算机导论课上听说了“刷&nbsp;LeetCode&nbsp;是进入大厂的敲门砖”,整个人就跟走火入魔了一样。他在宿舍门口贴了一张A4纸,上面写着:“正在&nbsp;DP,请勿打扰,否则&nbsp;Time&nbsp;Limit&nbsp;Exceeded。”日记本的扉页被他用黑色水笔加粗描了三遍:“Talk&nbsp;is&nbsp;cheap.&nbsp;Show&nbsp;me&nbsp;the&nbsp;code。”连宿舍聚餐,他都要给我们讲解:“今天的座位安排可以用回溯算法解决,但为了避免栈溢出,我建议用动态规划。来,这是状态转移方程:dp[i][j]&nbsp;代表第&nbsp;i&nbsp;个人坐在第&nbsp;j&nbsp;个位置的最优解。”我让他去楼下取个快递,他不直接去,非要在门口踱步,嘴里念念有词:“这是一个图的遍历问题。从宿舍楼(root)到驿站(target&nbsp;node),我应该用&nbsp;BFS&nbsp;还是&nbsp;DFS?嗯,求最短路径,还是广度优先好。”和同学约好出去开黑,他会提前发消息:“集合点&nbsp;(x,&nbsp;y),我们俩的路径有&nbsp;k&nbsp;个交点,为了最小化时间复杂度,应该在&nbsp;(x/2,&nbsp;y/2)&nbsp;处汇合。”有一次另一个室友低血糖犯了,让他帮忙找颗糖,他居然冷静地分析道:“别急,这是一个查找问题。零食箱是无序数组,暴力查找是&nbsp;O(n)。如果按甜度排序,我就可以用二分查找,时间复杂度降到&nbsp;O(log&nbsp;n)。”他做卫生也要讲究算法效率:“拖地是典型的岛屿问题,要先把连通的污渍区块都清理掉。倒垃圾可以用双指针法,一个指针从左往右,一个从右往左,能最快匹配垃圾分类。”现在我们宿舍的画风已经完全变了,大家不聊游戏和妹子,对话都是这样的:“你&nbsp;Two&nbsp;Sum&nbsp;刷了几遍了?”“别提了,昨天遇到一道&nbsp;Hard&nbsp;题,我连暴力解都想不出来,最后只能看题解。你呢?”“我动态规划还不行,总是找不到最优子结构。今天那道接雨水给我整麻了。”……LeetCode&nbsp;真的害了我室友!!!
老六f:编程嘉豪来了
AI时代还有必要刷lee...
点赞 评论 收藏
分享
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务