LeetCode-766. Toeplitz Matrix
原创
©著作权归作者所有:来自51CTO博客作者ReignsDu的原创作品,请联系作者获取转载授权,否则将追究法律责任
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:
In the above grid, the diagonals are:
"[9]", "[5, 5]", "[1, 1, 1]", "[2, 2, 2]", "[3, 3]", "[4]".
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]
.
Follow up:
- What if the matrix is stored on disk, and the memory is limited such that you can only load at most one row of the matrix into the memory at once?
- What if the matrix is so large that you can only load up a partial row into the memory at once?
题解:
class Solution {
public:
bool isToeplitzMatrix(vector<vector<int>>& matrix) {
int n = matrix.size();
int m = matrix[0].size();
for (int i = 0; i < m; i++) {
for (int j = i + 1, k = 1; j < m && k < n; j++, k++) {
if (matrix[k][j] != matrix[k - 1][j - 1]) {
return false;
}
}
}
for (int i = 1; i < n; i++) {
for (int j = i + 1, k = 1; j < n && k < m; j++, k++) {
if (matrix[j][k] != matrix[j - 1][k - 1]) {
return false;
}
}
}
return true;
}
};