6361对角线上的质数-leetcode周赛340第一道-Java

来源：第 340 场周赛 - 力扣（LeetCode）

一、问题描述

* 6361对角线上的质数 显示英文描述  给你一个下标从 0 开始的二维整数数组 nums 。返回位于 nums 至少一条 对角线 上的最大 质数 。
* 如果任一对角线上均不存在质数，返回 0 。注意：     如果某个整数大于 1 ，且不存在除 1 和自身之外的正整数因子，则认为该整数是一个质数。
* 如果存在整数 i ，使得 nums[i][i] = val 或者 nums[i][nums.length - i - 1]= val ，则认为整数 val 位于 nums 的一条对角线上。
* 在上图中，一条对角线是 [1,5,9] ，而另一条对角线是 [3,5,7] 。
* 示例 1：   输入：nums = [ [1,2,3],
* [5,6,7],
* [9,10,11]]
* 输出：11   解释：数字 1、3、6、9 和 11 是所有 "位于至少一条对角线上" 的数字。由于 11 是最大的质数，故返回 11 。
* 示例 2：   输入：nums = [[1,2,3],[5,17,7],[9,11,10]]
* 输出：17   解释：数字 1、3、9、10 和 17 是所有满足"位于至少一条对角线上"的数字。由于 17 是最大的质数，故返回 17 。
* 提示：     1 <= nums.length <= 300     nums.length == numsi.length     1 <= nums[i][j] <= 4*106

二、题解：

直接暴力：

class Solution {
    public int diagonalPrime(int[][] nums) {
        TreeSet<Integer> set = new TreeSet<>(new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                return -( o1 - o2);
            }
        });
        for (int i = 0; i < nums.length; i++) {
            set.add(nums[i][i]);
            set.add(nums[i][nums.length - 1 - i]);
        }

        while (set.size() != 0){
            int num = set.pollFirst();
            if (isDiagonalPrime(num)) return num == 1 ? 0 :num;//这里考虑1的情况

        }

        return 0;
    }

    private boolean isDiagonalPrime(int num) {

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

个人测试：

    @Test
    public void  testSolution(){

        int[][][] arr = {
                {{1,2,3},{4,5,6},{7,8,9}},
                {{1,2,3},{5,6,7},{9,10,11}},//11
                {{1,2,3},{5,17,7},{9,11,10}},//17
                {{1}},//1
                {{1,1,1},{1,1,1},{1,1,1}},//1
//                {{},{},{}},
        };

        for (int i = 0; i < arr.length; i++) {
            System.out.println(Arrays.deepToString(arr[i]) + ", result is " + diagonalPrime(arr[i]));
        }
    }