프로그래머스 - 월간 코드 챌린지 시즌1 > 삼각 달팽이

programmers.co.kr/learn/courses/30/lessons/68645?language=c

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

 

Solution

#include <stdio.h>
 
int dx[3] = { 0, 1, -1 };
int dy[3] = { 1, 0, -1 };
int a[1000][1000];
 
int* solution(int n) {
    
    int len = n * (n + 1) / 2;
    int* answer = (int*)calloc(len, sizeof(int));
    int dir = 0, cnt = 1;
    int ty = 0, tx = 0;
 
    while (1) {
        a[ty][tx] = cnt++;
        if (cnt > len) break;
 
        if (ty + dy[dir] == n || ty + dy[dir] < 0 ||
            tx + dx[dir] == n || tx + dx[dir] < 0 ||
            a[ty + dy[dir]][tx + dx[dir]] != 0) {
            dir = (++dir) % 3;
        }
 
        ty += dy[dir];
        tx += dx[dir];
    }
 
    int idx = 0;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n && a[i][j]!=0; j++) {
            answer[idx++] = a[i][j];
        }        
    }
 
    return answer;
}