[Leetcode] 766. Toeplitz Matrix 解题报告

题目：

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:

1. matrix will be a 2D array of integers.
2. matrix will have a number of rows and columns in range [1, 20].
3. 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;
    }
};