[프로그래머스] 정수를 나선형으로 배치하기

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

프로그래머스

---

 

 

Java

class Solution {
    public int[][] solution(int n) {
        int[][] answer = new int[n][n];
        
        // 순서대로 오른쪽, 아래, 왼쪽, 위
        int direction = 0;
        int[] dx = new int[]{0,1,0,-1};
        int[] dy = new int[]{1,0,-1,0};
        
        int curNum = 1;
        int row=0; int col=0;
        
        while (true){
            
            // 숫자를 채워넣는다. 
            answer[row][col] = curNum++;
            if(curNum > n*n) break;
            
            int nextRow = row + dx[direction%4];
            int nextCol = col + dy[direction%4];
            
            // 범위를 벗어났거나, 이미 숫자가 있으면 다음 위치를 갱신해주지 않고 방향만 바꿔준다.
            if(nextRow < 0 || nextRow >= n || nextCol<0 || nextCol >= n
              || answer[nextRow][nextCol] > 0){
                direction++;
                curNum--; 
            }else{
            	// 숫자를 채울 다음 위치를 갱신한다.
                row = nextRow;
                col = nextCol;
            }  
        }
        return answer;
    }
}