程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
 程式師世界 >> 編程語言 >> C語言 >> C++ >> C++入門知識 >> LeetCode Spiral Matrix II

LeetCode Spiral Matrix II

編輯:C++入門知識

LeetCode Spiral Matrix II


原題

將一個正方形矩陣螺旋著填滿遞增的數字。

注意點:

例子:

輸入: n = 3

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

解題思路

這道題跟 Spiral Matrix 正好相反,一個是螺旋著讀出數字,一個是螺旋著寫入數字,而且這道題還要簡單一點,因為形狀固定是正方形的,所以只要控制四條邊不斷向內縮進就可以了。考慮一下奇偶的情況,在奇數的時候要額外加一個中心點。

AC源碼

class Solution(object):
    def generateMatrix(self, n):
        """
        :type n: int
        :rtype: List[List[int]]
        """
        left = top = 0
        right = n - 1
        bottom = n - 1
        num = 1
        result = [[0 for __ in range(n)] for __ in range(n)]
        while left < right and top < bottom:
            for i in range(left, right):
                result[top][i] = num
                num += 1
            for i in range(top, bottom):
                result[i][right] = num
                num += 1
            for i in range(right, left, -1):
                result[bottom][i] = num
                num += 1
            for i in range(bottom, top, -1):
                result[i][left] = num
                num += 1
            left += 1
            right -= 1
            top += 1
            bottom -= 1
        if left == right and top == bottom:
            result[top][left] = num
        return result


if __name__ == "__main__":
    assert Solution().generateMatrix(5) == [[1, 2, 3, 4, 5], [16, 17, 18, 19, 6], [15, 24, 25, 20, 7],
                                            [14, 23, 22, 21, 8], [13, 12, 11, 10, 9]]
 

  1. 上一頁:
  2. 下一頁:
Copyright © 程式師世界 All Rights Reserved