开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第24天,点击查看活动详情
一、题目描述:
766. 托普利茨矩阵 - 力扣(LeetCode) (leetcode-cn.com)
给你一个 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 <= 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;
}