前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >LWC 68: 766. Toeplitz Matrix

LWC 68: 766. Toeplitz Matrix

作者头像
用户1147447
发布2019-05-26 19:40:21
4010
发布2019-05-26 19:40:21
举报
文章被收录于专栏:机器学习入门

Problem:

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:

  • matrix will be a 2D array of integers.
  • matrix will have a number of rows and columns in range [1, 20].
  • matrix[i][j] will be integers in range [0, 99].

思路: 对角线遍历,注意对角线的性质:当前元素为matrix[i][j], 下一元素为matrix[i+1][j+1]。

代码如下:

代码语言:javascript
复制
    public boolean isToeplitzMatrix(int[][] matrix) {
        int n = matrix.length;
        int m = matrix[0].length;

        for (int j = 0; j < m; ++j) {
            int i = 0;
            while (i + 1 < n && j + 1 < m) {
                if (matrix[i + 1][j + 1] != matrix[i][j]) return false;
                i ++;
                j ++;
            }
        }

        for (int i = 0; i < n; ++i) {
            int j = 0;
            while (i + 1 < n && j + 1 < m) {
                if (matrix[i + 1][j + 1] != matrix[i][j]) return false;
                i ++;
                j ++;
            }
        }
        return true;
    }

当然你可以直接遍历,只要判断每个元素所在对角线的下一个元素是否与当前元素相等即可。

Java版本:

代码语言:javascript
复制
    public boolean isToeplitzMatrix(int[][] matrix) {
        int n = matrix.length;
        int m = matrix[0].length;
        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;
    }

Python版本:

代码语言:javascript
复制
    def isToeplitzMatrix(self, matrix):
        """
        :type matrix: List[List[int]]
        :rtype: bool
        """
        n = len(matrix)
        m = len(matrix[0])

        for i in range(0, n - 1):
            for j in range(0, m - 1):
                if matrix[i][j] != matrix[i + 1][j + 1]: return False
        return True
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2018年01月21日,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档