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].

思路

本题我们先举个例子,加入存在一个3*3的Toeplitz矩阵,那么该矩阵如下所示:

1 2 3
5 1 2
9 5 1

我们发现,数字1对应的坐标如下所示:

(0,0)、(1,1)、(2,2)

数字2对应的坐标:

(0,1)、(1,2)

数字5对应的坐标:

(1,0)、(2,1)

很明显我们发现凡是数字相同的坐标都是x与y坐标自增1的
所以我们只要遍历完整个数组,并检查当前坐标(x,y)与(x+1,y+1)的关系即可。一旦发现不想等直接返回false,否则一直循环直至退出,返回true。

代码

class Solution {
public:
    bool isToeplitzMatrix(vector<vector<int>>& matrix) {
        int n = matrix.size();
        int m = matrix[0].size();
        for(int i=0;i<n-1;i++)
        {
            for(int j=0;j<m-1;j++)
            {
                if(matrix[i][j]!=matrix[i+1][j+1])
                    return false;
            }
        }
        return true;
    }
};

猜你喜欢

转载自blog.csdn.net/u010665216/article/details/79553305