首页 > 技术文章 > LeetCode 766. Toeplitz Matrix

tyty-Somnuspoppy 2018-08-02 16:58 原文

题意:判断给定矩阵的每一条主对角线上的元素是否都相同。

分析:

1、主对角线是从左上到右下。

2、同一对角线上的元素横纵坐标的差值相同,因此可以通过map来判定。

class Solution {
public:
    bool isToeplitzMatrix(vector<vector<int>>& matrix) {
        int r = matrix.size();
        int c = matrix[0].size();
        map<int, int> mp;
        for(int i = 0; i < r; ++i){
            for(int j = 0; j < c; ++j){
                if(!mp.count(i - j)){
                    mp[i - j] = matrix[i][j];
                }
                else{
                    if(mp[i - j] != matrix[i][j]) return false;
                }
            }
        }
        return true;
    }
};

  

推荐阅读