挑战498：对角线遍历，Simple Simulation 逐个击破

2023-11-26 02:06:20





## 破题：拥抱“简单模拟”

498题本质上是一个模拟题，考验我们对算法流程的把控能力。题目要求我们从给定的矩阵中提取出对角线元素，并按照从左上到右下的顺序排列。

面对这类模拟题，我们首先要做的就是理清思路，将问题分解成一个个小步骤。

1. **确定起始点：** 

    对角线遍历的起始点，可以选择矩阵的左上角或右上角。

2. **设定遍历方向：** 

    从起始点出发，我们可以朝右下或左下两个方向进行遍历。

3. **收集元素：** 

    在遍历过程中，我们将遇到的元素收集起来。

4. **移动起始点：** 

    当到达矩阵的边界或遍历完一条对角线时，我们需要移动起始点，继续下一个对角线的遍历。

## 逐个击破，算法解题

有了清晰的思路，我们就可以开始逐个击破，将抽象的问题转化为具体的步骤。

```python
def diagonal_traverse(matrix):
    """
    :type matrix: List[List[int]]
    :rtype: List[int]
    """
    if not matrix or not matrix[0]:
        return []

    m, n = len(matrix), len(matrix[0])
    result = []
    row, col = 0, 0
    # 从左上到右下遍历对角线
    while row < m and col < n:
        result.append(matrix[row][col])
        # 如果到达边界，则移动起始点
        if row == 0 or col == n - 1:
            if col == n - 1:
                row += 1
            else:
                col += 1
        # 否则，继续沿对角线遍历
        else:
            row -= 1
            col += 1

    # 从右下到左上遍历对角线
    while row < m and col >= 0:
        result.append(matrix[row][col])
        # 如果到达边界，则移动起始点
        if row == m - 1 or col == 0:
            if row == m - 1:
                col += 1
            else:
                row += 1
        # 否则，继续沿对角线遍历
        else:
            row += 1
            col -= 1

    return result