HelloCho

문제 :

이 문제는 #54에서 살펴본 문제와 동일한 문제로 동일한 알고리즘으로 해결할 수 있다. 다만 차이점이 있다면 #54 문제에서는 순회하면서 값을 추출했다면 이 문제에서는 나선형의 모양대로 값을 입력하면 된다. 

class Solution:
    def generateMatrix(self, n):
        res = [[0]* n for _ in range(n)]
        n_col = n
        n_row = n
        num_elements = n*n
        e = 0
        zone = 0
        data = 1
        
        while data <= num_elements:
            # move right 
            for i in range(zone, n_col - zone):
                res[zone][i] = data
                data +=1
            if data == num_elements+1:
                return res
            # move down
            for j in range(zone+1,n_row-zone):
                res[j][n_col-1-zone] = data
                data +=1
            if data == num_elements+1:
                return res
            # move left
            for i in range(n_col-1-zone,zone,-1):
                res[n_col-1-zone][i-1] = data
                data+=1
            if data == num_elements+1:
                return res
            # move up
            for j in range(n_row-2-zone,zone,-1):
                res[j][zone] = data
                data+=1
            if data == num_elements+1:
                return res
            zone +=1
        return res     

res에 빈 리스트를 만든 뒤 나선형으로 값을 1개씩 증가하면서 부여한다. 

결과 

28ms, 13.9mb로 통과 !