2614. 对角线上的质数

2614. 对角线上的质数

---

题目链接：2614. 对角线上的质数

代码如下：

class Solution {
public:
	int diagonalPrime(vector<vector<int>>& nums) {
		int res = 0;
		for (int i = 0;i < nums.size();i++) {
			if (isPrime(nums[i][i])) {
				res=max(res,nums[i][i]);
			}
			if (isPrime(nums[i][nums.size() - 1 - i])) {
				res = max(res, nums[i][nums.size() - 1 - i]);
			}
		}
		return res;
	}

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