小鱼刷leetcode---766. 托普利茨矩阵

34 阅读1分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第24天,点击查看活动详情

一、题目描述:

766. 托普利茨矩阵 - 力扣(LeetCode) (leetcode-cn.com)

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

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

示例 1:

image.png

输入: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:

image.png

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

 

提示:

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

进阶:

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

二、思路分析:

  • 思路是每行与下一行只需要有n-1个元素相同,每行更新也只需要改变首尾元素
  • 使用一个额外数组row保存上一行的元素;
  • 从上一行第2个元素开始与现在行的第一个元素比较,相同就继续,不同就返回false;
  • row数组在队头插入当前行的第一个元素,尾元素出栈,这样row就变成了现在行
  • 比较结束返回true;

三、AC 代码:

bool isToeplitzMatrix(vector<vector<int>>& matrix) 
    {
        deque<int> row = {};
        for(int i=0;i<matrix[0].size();i++)
            row.push_back(matrix[0][i]);
        for(int i=1;i<matrix.size();i++)
        {
            for(int j=1;j<matrix[0].size();j++)
                if(row[j-1] != matrix[i][j])
                    return false;
            row.push_front(matrix[i][0]);
            row.pop_back();
        }
        return true;
    }

范文参考:

C语言,依次遍历每一条对角线 - 托普利茨矩阵 - 力扣(LeetCode)