[프로그래머스] 삼각 달팽이 JAVA 문제 풀이 - 그림 설명 O

문제 설명

정수 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]

 

문제 풀이

문제의 그림을 행과 열로 바꾸어 표현해보겠다.

빨간 것이 좌표(x,y) 값이다.

표를 알맞는 방향으로 이동하면서 해당 칸에 차례대로 1++ 해서 증가시키면 배열로 반환하기 편할 것이다. 

아래 방향은 x만 증가, 오른쪽 이동은 y만 증가, 대각선 이동은 x,y값 둘다 감소인 것을 알수가 있다.

1️⃣ 순환하는 x좌표 값이 3으로 나누고 난 나머지가 0일 때, 아래로 이동한다. ( y값 고정, x값만증가) -> 이렇게 하려면 y값을 반복할때 시작점을 x좌표부터 해주면 될 것이다.

2️⃣ 나머지가 1일 때, y가 증가한다. (x값 고정, y값만 증가)

3️⃣ 나머지가 2일때, 대각선 이동이 발생한다.

 

** 1~n까지의 합은 n*(n+1)/2이다.

 

정답

class Solution {
    public int[] solution(int n) {
        // 세 방향 존재 -> 아래, 오른쪽, 대각선(x,y 둘다 감소)
        int x = -1; // 아래 이동은 x 좌표 감소
        int y = 0; // 오른쪽 이동은 y 좌표 증가
        int num = 1; // 삼각 달팽이 채우기 값
        // 삼각형
        int[][] tri = new int[n][n];
        // 삼각형의 크기 ( 1 ~ n 까지 합)
        int[] answer = new int[n*(n+1)/2]; 
        
        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--;
                }
                tri[x][y] = num++;
            }
        }
        // 결과 배열에 담기
        int index = 0;
		for(int i = 0; i < n; i++) {
			for(int j = 0; j < n; j++) {
				if(tri[i][j] == 0) break;
				answer[index++] = tri[i][j];
			}
		}
        return answer;
    }
}

 삼각 달팽이 삼각 달팽이 삼각 달팽이 삼각 달팽이 삼각 달팽이 삼각 달팽이 삼각 달팽이 삼각 달팽이 삼각 달팽이 삼각 달팽이 삼각 달팽이