삼각 달팽이

[문제 Link] https://programmers.co.kr/learn/courses/30/lessons/68645

문제 설명


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



제한사항


  • n은 1 이상 1,000 이하입니다.


입출력 예


n result
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]


입출력 예 설명



로직


문제를 해결하지 못하여, 정답을 보고 구현해보았다.

먼저 문제의 삼각형을 아래와 같이 모양을 바꾸었다.


위와 같이 모양을 바꾸면 숫자가 아래 방향, 오른쪽 방향, 왼쪽 위 대각선 방향으로 나뉘는 것을 알 수 있다. (아래의 그림)


위와 같이 모든 항목의 수의 갯수는 n(n+1)/2가 된다.

이제 마지막으로 숫자를 차례대로 좌표를 채우면 된다.

아래로 내려갈때는 x++ 해주고,

오른쪽으로 이동할 때는 y++ 해주고,

마지막으로 대각선 왼쪽 위로 이동할 때는 x–,y–를 해주면 된다.

전체 코드


using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;

class Solution
{
    public static int[] solution(int n) 
    {
        int[] answer = new int[(n*(n+1))/2];
        int[,] snail = new int[n,n];


        int x = -1, y = 0;
        int num = 1;

        for (int i = 0; i < n; i++)
        {
            for (int j = i; j < n; j++)
            {
                if (i % 3 == 0)
                {
                    x++;
                }
                else if (i % 3 == 1)
                {
                    y++;
                }
                else if (i % 3 == 2)
                {
                    x--;
                    y--;
                }
                snail[x,y] = num;
                num+=1;
            }
        }

        int k = 0;
        for (int i = 0; i < n; i++)
        {
            for (int j = 0; j < n; j++)
            {
                if (snail[i,j] == 0)
                    break;
                answer[k] = snail[i,j];
                k+=1;
            }
        }

        return answer;
    }


    public static void Main(string[] args)
    {
        int n = 6;
        int[] results = solution(n);
        foreach(var result in results)
            Console.WriteLine(result);
    }
}


결과