265.对角线上的质数

class Solution {
    public int diagonalPrime(int[][] nums) {
        int res=0;
        int n=nums.length;
        for(int i=0;i<n;i++){
            if(isPrime(nums[i][i]))
                res=Math.max(res,nums[i][i]);
            if(isPrime(nums[i][n-i-1]))
                res=Math.max(res,nums[i][n-i-1]);
        }
        return res;
    }
    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;
    }
}

class Solution(object):
    def diagonalPrime(self, nums):
        res=0
        n=len(nums)
        for i in range(n):
            if self.isPrime(nums[i][i]):
                res=max(res,nums[i][i])
            if self.isPrime(nums[i][n-i-1]):
                res=max(res,nums[i][n-i-1])
        return res
    def isPrime(self,num):
        if num==1:
            return False
        i=2
        while i*i<=num:
            if num%i==0:
                return False
            i+=1
        return True