[Algo Rhythm🕺💃] 프로그래머스 고득점 Kit 탑

문제 설명

수평 직선에 탑 N대를 세웠습니다. 모든 탑의 꼭대기에는 신호를 송/수신하는 장치를 설치했습니다. 발사한 신호는 신호를 보낸 탑보다 높은 탑에서만 수신합니다. 또한, 한 번 수신된 신호는 다른 탑으로 송신되지 않습니다.

예를 들어 높이가 6, 9, 5, 7, 4인 다섯 탑이 왼쪽으로 동시에 레이저 신호를 발사합니다. 그러면, 탑은 다음과 같이 신호를 주고받습니다. 높이가 4인 다섯 번째 탑에서 발사한 신호는 높이가 7인 네 번째 탑이 수신하고, 높이가 7인 네 번째 탑의 신호는 높이가 9인 두 번째 탑이, 높이가 5인 세 번째 탑의 신호도 높이가 9인 두 번째 탑이 수신합니다. 높이가 9인 두 번째 탑과 높이가 6인 첫 번째 탑이 보낸 레이저 신호는 어떤 탑에서도 수신할 수 없습니다.

 

송신 탑(높이)	수신 탑(높이)
5(4)	4(7)
4(7)	2(9)
3(5)	2(9)
2(9)	-
1(6)	-

 

맨 왼쪽부터 순서대로 탑의 높이를 담은 배열 heights가 매개변수로 주어질 때 각 탑이 쏜 신호를 어느 탑에서 받았는지 기록한 배열을 return 하도록 solution 함수를 작성해주세요.

 

제한 사항

• heights는 길이 2 이상 100 이하인 정수 배열입니다.

• 모든 탑의 높이는 1 이상 100 이하입니다.

• 신호를 수신하는 탑이 없으면 0으로 표시합니다.

 

나는 어떻게 풀었나

import java.util.Arrays;

class Solution {
    public int[] solution(int[] heights) {
        int n = heights.length;
        int[] answer = new int[n];
        Arrays.fill(answer, 0);
        
        for(int i = n - 1; i > 0; i--) {
            int h = heights[i];
            for(int j = i - 1; j >= 0; j--) {
                if(heights[j] > h){
                    answer[i] = j + 1;
                    break;
                }
            }
        }
        
        return answer;
    }
}

 

나는 왜 저렇게 풀었나

문제 유형이 스택/큐이긴 하지만 문제를 읽자마자 이건 완전 탐색 문제라고 생각했다. 왜냐하면 맨 오른쪽에 위치한 탑부터 순회하면서 그 탑보다 높은 탑만 찾으면 되는 문제였기 때문이다. 그래서 아래와 같은 알고리즘을 생각했다.

 

1. 탑의 개수 n이 길이인 배열을 선언하고 0으로 초기화한다.
2. 맨 오른쪽 탑부터 순회하면서 아래와 같은 과정을 거친다.
   1. 왼쪽에 있는 탑들을 순회하면서 더 높으면 배열에 저장한 뒤 2번 과정을 반복한다.

 

나는 무엇을 배웠나

Arryas javadoc을 읽다가 fill이라는 메소드를 발견했다. 앞으로 초기화할 때 애용해야겠다.

그리고 이 문제를 완탐으로 풀긴 했지만 스택/ 큐로는 어떻게 풀 수 있는지 궁금하다. 더 생각해봐야겠다.