[programmers] 삼각 달팽이 - Java

• 숫자는 위에서부터 아래, 오른쪽, 대각선 위 방향으로 순차적으로 증가 • 모듈로 연산(%)을 활용하여 방향 전환을 구현 • 2차원 배열을 1차원 배열로 변환하여 결과값 반환

DriedPollack

Dec 11, 2024

Contents

삼각 달팽이 solution.java 핵심 키워드 결론!

삼각 달팽이

문제 설명

정수 n이 매개변수로 주어집니다. 다음 그림과 같이 밑변의 길이와 높이가 n인 삼각형에서 맨 위 꼭짓점부터 반시계 방향으로 달팽이 채우기를 진행한 후, 첫 행부터 마지막 행까지 모두 순서대로 합친 새로운 배열을 return 하도록 solution 함수를 완성해주세요.

제한사항

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

입출력 예

n	result
4	`[1,2,9,3,10,8,4,5,6,7]`
5	`[1,2,12,3,13,11,4,14,15,10,5,6,7,8,9]`
6	`[1,2,15,3,16,14,4,17,21,13,5,18,19,20,12,6,7,8,9,10,11]`

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();
    }
}