2614.对角线上的质数

1.题目

2.思路

使用一个数组存储对角线数，按数大小将该数组排序，从最大的数开始验证是否是质数：是则输出；否则继续。如果一个质数都没有就直接输出0.

关键点：

（1）怎么判断质数：枚举用n对2~求余数，如果余数都不为零则为质数。

注意：“1”不是质数。

（2）如果都不是质数要输出0.

3.代码

class Solution {
public:
    bool isPrime(int num){
        if(num == 1){
            return false;
        }
        for(int i = 2;i<=sqrt(num);i++){
            if(num%i == 0){
                return false;
            }
        }
        return true;
    }
    int diagonalPrime(vector<vector<int>>& nums) {
        int n = nums.size();
        vector<int> f(2*n,0);
        for(int i = 0;i<n;i++){
            f[i] = nums[i][i];
        }
        for(int i = 0;i<n;i++){
            f[n+i] = nums[i][n-1-i];
        }
        cout<<endl;
        sort(f.begin(),f.end());
        int ans = 0;
        for(int i = 2*n-1;i>=0;i--){
            if(isPrime(f[i])){
                ans = f[i];
                break;
            }
        }
        return ans;
    }
};