leetcode766. Toeplitz Matrix

知识点总结

1 关于list的切片操作

a = range(::i)
#按步长i取list的值
a = range(4)
// a = [0,1,2,3]
b = a[::2]
// b = [0,2]

更多可参考该网址

点击打开链接

2 关于本题，思想还是很简单啦。矩阵对角线的元素都是a[i][j], a[i+1][j+1]...这种的，只需要判断他们是否相等就可以了。只不过我自己对于python的各种操作还是很不熟悉的。这里总结一下如何对list的行列去操作

1.
# a is a list
for i in range(len(a)-1):
    for j in range(len(zip(*a))-1):
        if a[i][j] != a[i+1][j+1]
            return False
return True

2.
for i in range(len(a)-1):
    for j in range(len(a[0])-1):
        ...
3.
for i, row in enumerate(a):
    for j, col in enumerate(row):
        ...

此外官方答案给出一种很简单的一行的写法，值得借鉴。（好像这样的话space complexity是O(1), 很有道理的样子，但我还不知道时间和空间复杂度是什么）

def Class(...):
    
    return all(i == 0 or j == 0 or a[i-1][j-1] == col in enumerate(a) for j, col in enumerate[row])