LeetCode-Python-498. 对角线遍历

给定一个含有 M x N 个元素的矩阵（M 行，N 列），请以对角线遍历的顺序返回这个矩阵中的所有元素，对角线遍历如下图所示。

 

示例:

输入:
[
 [ 1, 2, 3 ],
 [ 4, 5, 6 ],
 [ 7, 8, 9 ]
]

输出:  [1,2,4,7,5,3,6,8,9]

解释:

说明:

1. 给定矩阵中的元素总数不会超过 100000 。

 

思路：

由观察可以得知，一共有两种走法，向右上方走和向左下方走。（第一个元素也是向右上方走）

再由二维数组的特性可以得知，向右上方走实际上等于 x -= 1, y += 1, 向左下方走实际上等于 x+= 1, y -= 1。

 

然后再判断什么时候应该转弯：

1. 当向右上方走的时候，有两种情况会造成碰壁，因而需要转弯，

    CASE 1： 碰到上方的壁 （x无法再 - 1），但还没碰到右方的壁（y可以+1）

                     在这种情况下，下一步的坐标为y += 1， 比如上方示例图里的 1 -》 2。

   CASE 2： 碰到右方的壁 （y 不能 + 1）

                     在这种情况下，下一步的坐标为x += 1， 比如示例图里的 3 -》 6

2. 向左下方走同理：

   CASE1. 碰左壁但未碰下壁：x += 1

   CSSE2. 碰下壁：y += 1

class Solution(object):
    def findDiagonalOrder(self, matrix):
        """
        :type matrix: List[List[int]]
        :rtype: List[int]
        """
        # 向右上方走，x -= 1, y += 1
        # 向左下角走，x += 1, y -= 1
        m = len(matrix)
        if not m :
            return []
        n = len(matrix[0])
        if not n :
            return []

        cnt = 0
        x, y = 0, 0
        res = list()
        direction = "right"
        while(cnt < m * n):
            cnt += 1
            # print direction, x, y, matrix[x][y]
            res.append(matrix[x][y])
            if direction == "right":#向右上方走
                if x >= 1 and y < n - 1: # 不用调换方向
                    x -= 1
                    y += 1
                    continue
                else:
                    direction = "left" #调换方向
                    if x == 0 and y < n - 1: #碰上壁
                        y += 1
                    elif  y == n - 1: #碰右壁
                        x += 1
            else: # 向左下方走
                if x < m - 1 and y >= 1: # 不用调换方向
                    x += 1
                    y -= 1
                    continue
                else:
                    direction = "right" #调换方向
                    if x == m - 1: # 碰下壁
                        y += 1
                    elif y == 0 and x < m - 1: #碰左壁
                        x += 1 
            # print res
        return res