[Algorithm] Programmers :: 삼각 달팽이

https://school.programmers.co.kr/learn/courses/30/lessons/68645?language=cpp 

프로그래머스

 

vector<int> solution(int n) {
    vector<int> answer;
    vector<vector<int>> gameMap(1000, vector<int>(1000));
    vector<pair<int, int>> dir = { {-1,-1} , {1,0}, {0,1} };
 
    if (n == 1)
    {
        answer.push_back(1);
        return answer;
    }
 
    for (int i = 0; i < n; i++)
    {
        gameMap[i][0] = i + 1;
    }
 
    for (int i = 0; i < n; i++)
    {
        gameMap[n - 1][i] = i + n;
    }
 
    int curNum = gameMap[n - 1][n - 1];
    int moveCount = n - 2;
    int curX = n - 1;
    int curY = n - 1;
    int dirCount = 0;
 
    while (moveCount != 0)
    {
        if (dirCount == 3) dirCount = 0;
 
        int curDirY = dir[dirCount].first;
        int curDirX = dir[dirCount].second;
 
        for (int i = 0; i < moveCount; i++)
        {
            curY = curY + curDirY;
            curX = curX + curDirX;
 
            gameMap[curY][curX] = curNum + 1;
            curNum = curNum + 1;
        }
 
        moveCount--;
        dirCount++;
    }
 
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < i + 1; j++)
        {
            answer.push_back(gameMap[i][j]);
        }
    }
 
    return answer;
}

 

2차원배열에 달팽이모양으로 숫자찍어주는 문제이다. 

예재의 그림을 왼쪽정렬을 시키면 배열로 표현하기 좋게 그림이 나온다.

처음 n-1번만큼 수직,수평 이동시킨 후, moveCount가 0이 될때까지 방향을 바꿔가면서 moveCount만큼 찍어주면된다.

방향이 바뀔때마다 moveCount는 1씩 빠진다.

 

이런건 직접 노트에 그려보면 바로 감이 잡힌다.