LintCode Q185 Matrix Zigzag Traversal in Python

  • Jinhai ZHOU
  • 3 Minutes
  • 2017年1月30日
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
class Solution:
# @param: a matrix of integers
# @return: a list of integers
def printZMatrix(self, matrix):
# write your code here
res = []
m = len(matrix)
n = len(matrix[0])
limit = n + m - 1 - 1
level = 0
while level <= limit:
if level % 2 == 1:
for row in xrange(level + 1):
col = level - row
if row < m and col < n:
res.append(matrix[row][col])
level += 1
else:
for col in xrange(level + 1):
row = level - col
if row < m and col < n:
res.append(matrix[row][col])
level += 1
return res
知识共享许可协议
本作品采用知识共享署名-相同方式共享 4.0 国际许可协议进行许可。