【Leetcode】766. 托普利茨矩阵

QUESTION

easy

题目描述

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

给定一个 M x N 的矩阵，当且仅当它是托普利茨矩阵时返回 True。

示例 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]"上的元素不同。

说明

• matrix 是一个包含整数的二维数组。
• matrix 的行数和列数均在 [1, 20] 范围内。
• matrix[i][j] 包含的整数在 [0, 99] 范围内。

进阶:

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

---

SOLUTION

方法一

分成两部分来遍历，如图（遍历顺序如图中序号）：

class Solution {
public:
    bool isToeplitzMatrix(vector<vector<int>>& matrix) {
        int m = matrix.size(), n = matrix[0].size();
        for(int l = 0; l < n; l++){ //上三角
            int tmp = matrix[0][n-1-l];
            for(int i = 1; i <= l && i < m; i++){
                if(matrix[i][n-1-l+i] != tmp) return false;
            }
        }
        for(int l = 1; l < m; l++){ //下三角
            int tmp = matrix[l][0];
            for(int i = l + 1; i < m && i-l < n; i++){
                if(matrix[i][i-l] != tmp) return false;
            }
        }
        return true;
    }
};

方法二

实际上也不用那么麻烦，我们可以按正常方法顺序遍历，只要每次和右下的元素比较是否相同即可。（最后一行和最后一列是可以不用遍历的，想想为什么）

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

方法三

转自 Grandyang 的博客，建议前往阅读

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