3115. 质数的最大距离

给你一个整数数组 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 * 105
• 1 <= nums[i] <= 100
• 输入保证 nums 中至少有一个质数。

我的解题，找到数组中的质数位置， 然后双指针，找到最前和最后的质数位置即可

官方的解题，列出100内的所有质数，从左向后遍历，找到首元素， 然后遍历到最后，算出最大长度

public class Q3115 {

    public int maximumPrimeDifference(int[] nums) {

        for (int i = 0; i < nums.length; i++) {
            if (isPrime(nums[i])) {
                nums[i] = -1;
            }
        }
        int max = 0;
        for (int head = 0, tail = nums.length - 1; tail - head >= 1; ) {
            if (nums[head] == -1 && nums[tail] == -1) {
                max = tail - head;
                break;
            }
            if (nums[head] != -1) {
                head++;
            } else if (nums[tail] != -1) {
                tail--;
            }
        }
        return max;
    }

    public static boolean isPrime(int num) {
        if (num <= 1) {
            return false;
        }

        for (int i = 2; i * i <= num; i++) {
            if (num % i == 0) {
                return false;
            }
        }
        return true;
    }

    public static void main(String[] args) {
        Q3115 q3115 = new Q3115();
        int[] nums = {2, 2};
        System.out.println(q3115.maximumPrimeDifference(nums));
    }

}