삼각 달팽이

삼각 달팽이

정수 n이 매개변수로 주어집니다. 다음 그림과 같이 밑변의 길이와 높이가 n인 삼각형에서 맨 위 꼭짓점부터 반시계 방향으로 달팽이 채우기를 진행한 후, 첫 행부터 마지막 행까지 모두 순서대로 합친 새로운 배열을 return 하도록 solution 함수를 완성해주세요.

입출력 예

n	return
4	[1,2,9,3,10,8,4,5,6,7]
5	[1,2,12,3,13,11,4,14,15,10,5,6,7,8,9]
6	[1,2,15,3,16,14,4,17,21,13,5,18,19,20,12,6,7,8,9,10,11]

---

풀이

dfs 로 풀려다가 실패하고...

def solution(n):
    answer = []
    for i in range(1,n+1): # 리스트 만들기
        answer.append([0] * i)
    idx = 0
    def dfs(x,y,idx):
        start = answer[x][y]
        for i in range(len(answer)):# 첫값 채우기
            for j in range(len(answer[i])):
                #if 
                answer[i][0] = j+1

        for i in range(n): # 마지막 리스트 값 채우기
            answer[-1][i] = n + i


        for i in range(n-2,0,-1): # 리스트의 마지막값들 채우기
            answer[i][-1] = answer[i+1][-1] + 1
        print(answer)
        dfs(x+1,y,idx)
    
    dfs(0,0,0)
    
    print(answer)
    #return answer

정답 풀이

수학적 사고방식 필요
배열 끝에 다다를 때, 방향 전환 필요

def solution(n):
   answer = []
  
  for i in range(1,n+1):
      answer.append(i*[0])
  print(answer)
  x, y = -1, 0 
  ## 처음엔 무조건 아래로 향하게 해야 하므로 x=-1로 초기화
  num = 1

  for i in range(n): # 방향
      for j in range(i, n): # 삼각형 모양의 배열을 만들기 위해 i부터 시작
          
          # 삼각형이라서 3
          if i % 3 == 0: # 하
              x += 1
              #print("나머지0", x,y,i,j)
          elif i % 3 == 1: # 우
              y += 1
              #print("나머지1", x,y,i,j)
          else: # 상
              x -= 1
              y -= 1
              #print("나머지2", x,y,i,j)
          answer[x][y] = num
          print(x,y, answer,"x좌표 넘어가")
          num += 1

  return sum(answer, [])
  

⚡️ TIL

sum(answer,[])

2차원리스트 answer 를 1차원리스트로 평탄화

[[1], [2, 3], [4, 5, 6], [7, 8, 9, 10]]

=> [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]