python leetcode 54. Spiral Matrix 59. Spiral Matrix II

Spiral Matrix

class Solution(object):
    def spiralOrder(self, matrix):
        """
        :type matrix: List[List[int]]
        :rtype: List[int]
        """
        res=[]
        def circle(mat1):
            if mat1==[]:
                return 
            m = len(mat1)
            n = len(mat1[0])
            for j in range(n):
                res.append(mat1[0][j])
            for i in range(1,m-1):
                res.append(mat1[i][n-1])
            for j in range(n-1,-1,-1):
                if m==1:
                    break
                res.append(mat1[m-1][j])
            for i in range(m-2,0,-1):
                if n==1:
                    break
                res.append(mat1[i][0])
            if m<3 or n < 3:
                m1=[]
            else:
                m1=[[mat1[i][j] for j in range(1,n-1)] for i in range(1,m-1)]
            circle(m1)
            return  
        circle(matrix)
        return res

Spiral Matrix II

class Solution(object):
    def generateMatrix(self, n):
        """
        :type n: int
        :rtype: List[List[int]]
        """
        res=[[0]*n for i in range(n)]
        def generate(start,end,first):
            if end <= start:
                return 
            for j in range(start,end):
                res[start][j]=first
                first+=1
            for i in range(start+1,end-1):
                res[i][end-1]=first
                first+=1
            for j in range(end-1,start-1,-1):
                if res[end-1][j]==0:
                    res[end-1][j]=first
                    first+=1
            for i in range(end-2,start,-1):
                res[i][start]=first
                first+=1 
            generate(start+1,end-1,first)
            return
        generate(0,n,1)
        return res