托普利茨矩阵(力扣第766题)

766. 托普利茨矩阵

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

给定一个 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]"上的元素不同。

实现:

   public boolean isToeplitzMatrix(int[][] matrix) {

        int m = matrix.length;
        int n = matrix[0].length;

        if (m == 0 || n == 0 || matrix == null){
            return true;
        }
        for (int i = 0; i < n; i++) {

            int cur = matrix[0][i];

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

        for (int i = 1; i < m; i++) {
            int cur = matrix[i][0];
            for (int j = i,k = 0; j < m && k < n; j++,k++) {
                if (matrix[j][k] != cur){
                    return false;
                }
            }
        }
        return true;
    }
posted @ 2020-12-01 19:53  有心有梦  阅读(89)  评论(0编辑  收藏  举报