LeetCode 3115.质数的最大距离：质数判断

【LetMeFly】3115.质数的最大距离：质数判断

力扣题目链接：https://leetcode.cn/problems/maximum-prime-difference/

给你一个整数数组 nums。

返回两个（不一定不同的）质数在 nums 中下标的 最大距离。

示例 1：
输入： nums = $4,2,9,5,3$

输出： 3

解释： nums[1]、nums[3] 和 nums[4] 是质数。因此答案是 |4 - 1| = 3。

示例 2：
输入： nums = $4,8,2,8$

输出： 0

解释： nums[2] 是质数。因为只有一个质数，所以答案是 |2 - 2| = 0。

提示：

1 <= nums.length <= 3 * 10⁵
1 <= nums[i] <= 100
输入保证 nums 中至少有一个质数。

解题方法：质数判断

如何判断一个整数是否为质数？

对于一个整数 n n n，如果 n < 2 n\lt 2 n<2，则 n n n不是质数。

使用变量 i i i从 2 2 2到 n \sqrt{n} n 遍历，若对于某个 i i i有 n % i = = 0 n\% i==0 n%i==0，则 n n n不是质数。

否则， n n n为质数。

接着遍历整数数组，使用两个变量即可确定出第一个质数和最后一个质数。二者相减即为答案。

时间复杂度 O ( l e n ( n u m s ) × max ⁡ ( n u m s ) ) O(len(nums)\times\sqrt{\max(nums)}) O(len(nums)×max(nums) )
空间复杂度 O ( 1 ) O(1) O(1)

AC代码

C++

cpp 复制代码

class Solution {
private:
    inline bool isPrime(int n) {
        if (n == 1) {
            return false;
        }
        for (int i = 2; i * i <= n; i++) {
            if (n % i == 0) {
                return false;
            }
        }
        return true;
    }
public:
    int maximumPrimeDifference(vector<int>& nums) {
        int m = 10000000, M = -1;
        for (int i = 0; i < nums.size(); i++) {
            if (isPrime(nums[i])) {
                m = min(m, i);
                M = max(M, i);
            }
        }
        return M - m;
    }
};

Go

go 复制代码

package main

func max(a int, b int) int {
    if a > b {
        return a
    }
    return b
}

func min(a int, b int) int {
    if a < b {
        return a
    }
    return b
}

func isPrime(n int) bool {
    if n == 1 {
        return false
    }
    for i := 2; i * i <= n; i++ {
        if n % i == 0 {
            return false
        }
    }
    return true
}

func maximumPrimeDifference(nums []int) int {
    M, m := -1, 10000000
    for i := 0; i < len(nums); i++ {
        if isPrime(nums[i]) {
            M = max(M, i)
            m = min(m, i)
        }
    }
    return M - m
}

Java

java 复制代码

class Solution {
    private boolean isPrime(int n) {
        if (n == 1) {
            return false;
        }
        for (int i = 2; i * i <= n; i++) {
            if (n % i == 0) {
                return false;
            }
        }
        return true;
    }

    public int maximumPrimeDifference(int[] nums) {
        int m = 10000000, M = -1;
        for (int i = 0; i < nums.length; i++) {
            if (isPrime(nums[i])) {
                m = Math.min(m, i);
                M = Math.max(M, i);
            }
        }
        return M - m;
    }
}

Python

python 复制代码

from typing import List
from math import sqrt

class Solution:
    def isPrime(self, n: int) -> bool:
        if n == 1:
            return False
        for i in range(2, int(sqrt(n)) + 1):
            if n % i == 0:
                return False
        return True
    
    def maximumPrimeDifference(self, nums: List[int]) -> int:
        m, M = 10000000, -1
        for i in range(len(nums)):
            if self.isPrime(nums[i]):
                m, M = min(m, i), max(M, i)
        return M - m

同步发文于CSDN和我的个人博客，原创不易，转载经作者同意后请附上原文链接哦~

Tisfy：https://letmefly.blog.csdn.net/article/details/140121329