프로그래머스[Java] - 삼각 달팽이

programmers.co.kr/learn/courses/30/lessons/68645

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

 

 

 

 

코드

class Solution {
	public static int[] solution(int n) {
		if(n == 1) {
			return new int[] {1};
		}

		int answerLen = 3;
		for(int i=3; i<=n; i++) {
			answerLen += i;
		}
		int[] answer = new int[answerLen];

		int[][] arr = new int[n][n];
		int count = 1;	// 배열 값
		int row = -1;	// 세로
		int col = 0;	// 가로
		int answerIndex = 0;	

		while(true) {
			// ↓ 방향(row 증가)
			for(int i=0; i<n; i++) {
				row += 1;
				arr[row][col] = count;
				count ++;
			}
			n --;
			if(n == 0) {
				break;
			}

			// → 방향(col 증가)
			for(int i=0; i<n; i++) {
				col += 1;
				arr[row][col] = count;
				count ++;
			}
			n --;
			if(n == 0) {
				break;
			}

			// ↖ 방향(row, col 감소)
			for(int i=0; i<n; i++) {
				row -= 1;
				col -= 1;
				arr[row][col] = count;
				count ++;
			}
			n --;
			if(n == 0) {
				break;
			}
		}

		for(int i=0; i<arr.length; i++) {
			for(int j=0; j<arr[i].length; j++) {
				if(arr[i][j] == 0) {	// 각 row에서 값이 없을경우 => break
					break;
				}
				answer[answerIndex] = arr[i][j];
				answerIndex ++;
			}
		}
		return answer;
	}

}

풀이

기존에 사각형 달팽이 문제와 유사한 삼각형 달팽이 문제이다.

 

이 삼각형 달팽이 문제는 규칙을 찾는것이 우선이다. 사각형 달팽이와 비슷한데 내가 찾은 규칙은 다음과 같다.

 

n=4 일때를 기준으로 보자.

 

위와 같은 삼각형을 

 

위와 같은 이등변 삼각형으로 보고, 각 방향대로 값을 채워 배열에 담아서 리턴해주면 된다.

 

↓ , → , ↖ 각 방향으로 갈 때마다, 반복되는 횟수가 1회씩 줄어든다.

 

4회 반복 ↓ 

3회 반복  → 

2회 반복 ↖

1회 반복 ↓

0회 반복 (끝)

 

따라서, 입력받은 n이 0이 될때까지, ↓ , → , ↖ 방향으로 진행하면서 2차원 배열에 값을 채우고,

각 단계가 끝날때마다 n을 감소시킨다.

 

각 단계마다 row, col 값이 증가하거나 감소하는데, 규칙은 다음과 같다.

 

↓ 방향:  row(행) 1씩 증가

			// ↓ 방향(row 증가)
			for(int i=0; i<n; i++) {
				row += 1;
				arr[row][col] = count;
				count ++;
			}

 

→ 방향:  col(열) 1씩 증가

			// → 방향(col 증가)
			for(int i=0; i<n; i++) {
				col += 1;
				arr[row][col] = count;
				count ++;
			}

 

↖ 방향: row(행), col(열) 1씩 감소

			// ↖ 방향(row, col 감소)
			for(int i=0; i<n; i++) {
				row -= 1;
				col -= 1;
				arr[row][col] = count;
				count ++;
			}

 

위와 같은 규칙에 따라 각 단계의 for문에서 row, col 값을 변경시켜준다.

 

* 처음 row의 값을 -1로 정한 이유는, 만약 row 값을 0으로 지정할경우,

첫 for문이 끝난 뒤의 row 값은 n+1이 되어 인덱스의 범위를 벗어나게된다.

 

* 처음에 리턴할 배열 answer의 길이를 구해야하는데, 이것도 입력받는 n에 따라 규칙이 정해진다.

n , 숫자의 총 갯수에 따른 규칙은 다음과 같다.

n	숫자의 갯수(배열의 길이)
2	3   (+0)
3	6   (+3)
4	10  (+4)
5	15  (+5)
6	21  (+6)

 

위 규칙에 따라 배열의 길이는 i가 3부터, 입력받은 n까지 i값(3, 4, 5, ...)을 더해주면 된다.

		int answerLen = 3;
		for(int i=3; i<=n; i++) {
			answerLen += i;
		}