[programmers] 삼각 달팽이 - Java

삼각 달팽이

문제 설명

제한사항

• n은 1 이상 1,000 이하입니다.

입출력 예

solution.java

import java.util.*;

class Solution {
    public int[] solution(int n) {
        int[][] triangle = new int[n][n];
        int num = 1; // 채워넣을 숫자
        int x = -1, y = 0; // 시작 위치

        // 삼각 달팽이 채우기
        for (int i = 0; i < n; i++) {
            for (int j = i; j < n; j++) {
                if (i % 3 == 0) { // 아래로 이동
                    x++;
                } else if (i % 3 == 1) { // 오른쪽으로 이동
                    y++;
                } else if (i % 3 == 2) { // 대각선 위로 이동
                    x--;
                    y--;
                }
                triangle[x][y] = num++;
            }
        }
        
        // result 생성
        List<Integer> result = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            for (int j = 0; j <= i; j++) {
                result.add(triangle[i][j]);
            }
        }

        return result.stream().mapToInt(Integer::intValue).toArray();
    }
}

핵심 키워드

• i % 3 == 0, i % 3 == 1, i % 3 == 2 의 조건을 통해 어느 방향으로 숫자가 증가할지 결정한다.

결론!