[ 프로그래머스 ] 삼각 달팽이

문제

---

https://school.programmers.co.kr/learn/courses/30/lessons/68645

풀이

---

밑변의 길이와 높이가 n인 삼각형에서 맨 위 꼭짓점부터 반시계 방향으로 달팽이 채우기를 진행한 후, 첫 행부터 마지막 행까지 모두 순서대로 합친 새로운 배열을 return하는 문제로, 단순하게 각 배열의 맨 앞자리, 마지막 배열, 각 배열의 맨 뒷자리를 순서대로 채운 배열을 return한다.

#include <string>
#include <vector>

using namespace std;

 
vector<int> solution(int n)
{
    vector<vector<int>> MAP(n + 1, vector<int>(n + 1));
    vector<int> answer;
    int Max_Num = (n * (n + 1)) / 2;
    int Top = 1;
    int Bottom = n;
    int Left = 1;
    int Right = 0;
    int Num = 1;
    int State = 0;
    while (Num <= Max_Num)
    {
        if (State == 0)
        {
            for (int i = Top; i <= Bottom; i++) MAP[i][Left] = Num++;
            Top++;
            Left++;
            State = 1;
        }
        else if (State == 1)
        {
            for (int i = Left; i <= Bottom - Right; i++) MAP[Bottom][i] = Num++;
            Bottom--;
            State = 2;
        }
        else if (State == 2)
        {
            for (int i = Bottom; i >= Top; i--) MAP[i][i - Right] = Num++;
            Right++;
            Top++;
            State = 0;
        }
    }
 
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= i; j++)
        {
            answer.push_back(MAP[i][j]);
        }
    }
    
    return answer;
}