LeetCode 766. 托普利茨矩阵

SakuraTears

发布于 2022-01-13 14:40:34

1410

发布于 2022-01-13 14:40:34

文章被收录于专栏：从零开始的Code生活

题目

给你一个 m x n 的矩阵 matrix 。如果这个矩阵是托普利茨矩阵，返回 true ；否则，返回 false 。

如果矩阵上每一条由左上到右下的对角线上的元素都相同，那么这个矩阵是托普利茨矩阵。

示例 1：

输入：matrix = [[1,2,3,4],[5,1,2,3],[9,5,1,2]] 输出：true 解释：在上述矩阵中, 其对角线为: "[9]", "[5, 5]", "[1, 1, 1]", "[2, 2, 2]", "[3, 3]", "[4]"。各条对角线上的所有元素均相同, 因此答案是 True 。

示例 2：

输入：matrix = [[1,2],[2,2]] 输出：false 解释：对角线 "[1, 2]" 上的元素不同。

提示：

m == matrix.length n == matrix[i].length 1 <= m, n <= 20 0 <= matrixi <= 99

进阶：

如果矩阵存储在磁盘上，并且内存有限，以至于一次最多只能将矩阵的一行加载到内存中，该怎么办？如果矩阵太大，以至于一次只能将不完整的一行加载到内存中，该怎么办？

思路

比较每一条对角线元素是否相同，从数组的横向元素和纵向元素为起点判断if (matrix[m][n] != matrix[m + 1][n + 1]) return false;即可

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

    bool isToeplitzMatrix(vector<vector<int>>& matrix) {
        for (int i = 0; i < matrix[0].size(); i++) {
            if (judge(matrix, 0, i) == false) return false;
        }
        for (int i = 0; i < matrix.size(); i++) {
            if (judge(matrix, i, 0) == false) return false;
        }
        return true;
    }
};

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2021年02月22日，如有侵权请联系 cloudcommunity@tencent.com 删除

matrix