题目:
A matrix is Toeplitz if every diagonal from top-left to bottom-right has the same element.
Now given an M x N
matrix, return True
if and only if the matrix is Toeplitz.
Example 1:
Input: matrix = [[1,2,3,4],[5,1,2,3],[9,5,1,2]] Output: True Explanation: 1234 5123 9512 In the above grid, the diagonals are "[9]", "[5, 5]", "[1, 1, 1]", "[2, 2, 2]", "[3, 3]", "[4]", and in each diagonal all elements are the same, so the answer is True.
Example 2:
Input: matrix = [[1,2],[2,2]] Output: False Explanation: The diagonal "[1, 2]" has different elements.
Note:
matrix
will be a 2D array of integers.matrix
will have a number of rows and columns in range[1, 20]
.matrix[i][j]
will be integers in range[0, 99]
.
思路:
逐个检查就可以了,不过需要注意控制边界条件,以防止越界访问。
代码:
class Solution {
public:
bool isToeplitzMatrix(vector<vector<int>>& matrix) {
int row_num = matrix.size(), col_num = matrix[0].size(), value = 0;
for (int c = 0; c < col_num; ++c) {
value = matrix[0][c];
for (int r = 1; r < row_num && r + c < col_num; ++r) {
if (matrix[r][r + c] != value) {
return false;
}
}
}
for (int r = 1; r < row_num; ++r) {
value = matrix[r][0];
for (int c = 1; c < col_num && r + c < row_num; ++c) {
if (matrix[r + c][c] != value) {
return false;
}
}
}
return true;
}
};