leetcode--矩阵遍历与搜索

最新推荐文章于 2023-12-18 10:00:00 发布

原创最新推荐文章于 2023-12-18 10:00:00 发布 · 412 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

刷题专栏收录该内容

9 篇文章

订阅专栏

本文探讨了螺旋矩阵的实现及其两种变体（螺旋矩阵I和II），以及如何在二维矩阵中进行高效搜索。通过实例展示了Solution类中的算法，涵盖了递归和迭代方法，适用于数据结构和算法初学者或进阶者。

54. 螺旋矩阵

class Solution(object):
    def spiralOrder(self, matrix):
        res = []
        row, col= len(matrix)-1, len(matrix[0])-1
        i, j = 0, 0

        while i<=row and j <=col:
            if i == row:
                for x in range(j, col+1):
                    res.append(matrix[row][x])
                return res
            
            if j == col:
                for x in range(i, row+1):
                    res.append(matrix[x][col])
                return res

            for x in range(j, col+1):
                res.append(matrix[i][x])
            
            for x in range(i+1, row+1):
                res.append(matrix[x][col])


            for x in range(col-1, j-1, -1):
                res.append(matrix[row][x])
            
            for x in range(row-1, i, -1):
                res.append(matrix[x][j])
            i += 1
            j += 1
            row -= 1
            col -= 1
        return res

59. 螺旋矩阵 II

class Solution(object):
    def generateMatrix(self, n):
        """
        :type n: int
        :rtype: List[List[int]]
        """
        matrix = [[0] * n for _ in range(n)]
        i, j = 0, 0
        a, b=  n -1, n-1
        number = 1
        res = []
        while i <= a and j <= b:
            if i == a and j == b:
                matrix[i][j] = number
                return matrix
            
            for k in range(j, b+1):
                matrix[i][k] = number
                number += 1

            for k in range(i+1, a+1):
                matrix[k][b]  = number
                number += 1

            for k in range(b-1, j-1, -1):
                matrix[a][k] = number
                number += 1


            for k in range(a-1, i, -1):
                matrix[k][j] = number
                number += 1

            i += 1
            j += 1
            a -= 1
            b -= 1 

        return matrix

74. 搜索二维矩阵
 240. 搜索二维矩阵 II

class Solution(object):
    def searchMatrix(self, matrix, target):
        m,n = len(matrix)-1, len(matrix[0])-1
        
        i, j = m, 0

        while i>=0 and j<=n:
            if matrix[i][j] == target:
                return True
            elif matrix[i][j] > target:
                i -= 1
            else:
                j += 1
        
        return False