[프로그래머스] 삼각달팽이 for JAVA

 

문제설명

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

제한사항

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

 

문제풀이

class Solution {
	public int[] solution(int n) {
		int[][] arr = new int[n][n];
		int num = 1; // 배열에 들어갈 값
		int max = n * (n + 1) / 2; // 최대값
		int row = 0; // 행
		int col = 0; // 열
		arr[0][0] = 1;
		while (num < max) {
			// 왼쪽
			while (row + 1 < n && arr[row + 1][col] == 0)
				arr[++row][col] = ++num;
			// 아래쪽
			while (col + 1 < n && arr[row][col + 1] == 0)
				arr[row][++col] = ++num;
			// 오른쪽
			while (col - 1 > 0 && row - 1 > 0 && arr[row - 1][col - 1] == 0)
				arr[--row][--col] = ++num;
		}
		// 결과양식으로 변환
		int[] answer = new int[num];
		int idx = 0;
		for (int i = 0; i < n; i++) {
			for (int j = 0; j < n; j++) {
				if (arr[i][j] == 0)
					continue;
				answer[idx++] = arr[i][j];
			}
		}
		return answer;
	}
}

 

문제 바로가기

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