[c++] 프로그래머스 - 삼각 달팽이(Level 2)

문제

코딩테스트 연습 - 삼각 달팽이

코드

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


using namespace std;

vector<int> solution(int n) {
    vector<int> answer;
    
    int num = 1;
    int move_cnt = n;
    
    int arr[1001][1001]={0,};
    int x=0, y=0;
    
    for(int i=0; i<n*n; i++){
        if(arr[x][y]!=0) break;
        switch(i%4){
            case 0: //아래로 이동
                for(int k=0; k<move_cnt; k++){
                    arr[x][y] = num;
                    num++;
                    x++;
                }
                x--;
                y++;
                move_cnt--;
                break;
                
            case 1: //우측으로 이동
                for(int k=0; k<move_cnt; k++){
                    arr[x][y] = num;
                    num++;
                    y++;
                }
                y--;
                x--;
                move_cnt--;
                break;
                
            case 2: //위쪽으로 이동
                for(int k=0; k<move_cnt; k++){
                    arr[x][y] = num;
                    num++;
                    x--;
                }
                x++;
                y--;
                move_cnt--;
                break;
                
            case 3: //좌측으로 이동
                x++;
                int move_x = x, move_y = y;
                
                while(true){
                    if(arr[move_x][move_y]!=0)break;
                    move_y--;
                }
                y = move_y+1;
                break;
        }
    }
    
    for(int i=0; i<1001; i++){
        for(int k=0; k<1001; k++){
            if(arr[i][k]!=0){
                answer.push_back(arr[i][k]);
            }
        }
    }
    return answer;
}

 

풀이(40분)

1913번: 달팽이

위와 같은 기초적인 달팽이 문제에서 조금 더 생각을 해야 했던 문제였습니다. 저처럼 상하좌우 이동 방향마다 구현했을 시 좌측으로 갈 때 특히 신경 쓸 부분이 많았는데 아래 이미지처럼 좌측 이동시 아래로 한 칸 이동 후 젤 좌측 빈칸으로 가서 다시 시작합니다. 이 부분만 신경 써 준다면 다른 부분은 크게 어렵지 않았습니다.