(프로그래머스 c++)삼각 달팽이

문제 설명

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

---

제한사항

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

 

for문과 switch문을 이용하여 2차원 배열에 삼각 달팽이를 만든다.

 

수를 채워 나갈 때 방향이 꺽이는 횟수가 n과 같으므로 0 ~ n까지 for문을 반복.

state 변수를 이용하여 0 = 아래로 채우는 방향, 1 = 오른쪽으로 채우는 방향, 2 = 위로 채우는 방향을 설정한다.

 

switch문에서 state를 이용하여 각 케이스 별로 수를 채워나간다.

i ~ n까지 반복.

 

0인 경우 arr[x++][y] = num++

빠져 나가기 전 state = 1, x--, y++을 하여 오른쪽으로 채워 나갈 때의 좌표를 설정해준다.

 

1인 경우 arr[x][y++] = num++

빠져 나가기 전 state = 2, x--, y -= 2를 하여 위로 채워 나갈 때의 좌표를 설정해준다.

 

2인 경우 arr[x--][y--] = num++

빠져 나가기 전 state = 0, x += 2, y++을 하여 다시 아래쪽으로 채워 나갈 때의 좌표를 설정해준다.

 

answer에는 이중 for문을 이용해 차례대로 넣어주면 된다.

#include <string>
#include <vector>
#include <iostream>

using namespace std;

vector<int> solution(int n) {
    vector<int> answer;
    int arr[1001][1001] = {0, };
    int x = 0, y = 0, state = 0, num = 1;
    
    for(int i = 0; i < n; i++)
    {
        switch(state)
        {
            case 0:
                for(int j = i; j < n; j++)
                    arr[x++][y] = num++;
                x--;
                y++;
                state = 1;
                break;
            case 1:
                for(int j = i; j < n; j++)
                    arr[x][y++] = num++;
                x--;
                y -= 2;
                state = 2;
                break;
            case 2:
                for(int j = i; j < n; j++)
                    arr[x--][y--] = num++;
                x += 2;
                y++;
                state = 0;
                break;
        }
    }
    
    for(int i = 0; i < n; i++)
    {
        for(int j = 0; j <= i; j++)
            answer.push_back(arr[i][j]);
    }
    return answer;
}