탑 (프로그래머스) [JAVA]

 

문제

수평 직선에 탑 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으로 표시합니다.

입출력 예

heights	return
[6,9,5,7,4]	[0,0,2,2,4]
[3,9,9,3,5,7,2]	[0,0,0,3,3,3,6]
[1,5,3,6,7,6,5]	[0,0,2,0,0,5,6]

 

풀이 (1)

단순하게 배열로 2중 for문을 사용해서 풀었다.

 

코드 (1)

class Solution {
    public int[] solution(int[] towers) {
        int[] answer = new int[towers.length];
        lazer(towers, answer);
        return answer;
    }
    
    public void lazer(int[] tower, int[] answer) {
    	for(int i=tower.length-1; i>=0; i--) { 	  		
    		for(int j=i; j>=0; j--) {
    			if(tower[i] < tower[j]) {
    				answer[i] = j+1;
    				break;
    			}
    		}
    		
    	}
    }
}

 

풀이 (2)

스택을 사용해서 풀었을때는, 스택에 인덱스와 높이가 담긴 배열을 넣으면서 탐색하였다.
스택이 비어있지 않을 때 맨 위에 있는 노드가 들어갈 노드(노드 왼쪽 타워)보다 크면 그 타워가 레이저를 맞고,
작으면 그 타워는 pop한뒤 나머지 노드들을 탐색하는 방법이다.

 

코드(2)

class Solution {
    public int[] solution(int[] towers) {
        Stack<int[]> st = new Stack<int[]>();
        int[] answer = new int[towers.length];

        for (int i = 0; i < towers.length; i++) {
            int[] tower = new int[2];
            tower[0] = i+1; tower[1] = towers[i];
            int receiveIdx = 0;

            while (!st.isEmpty()) {
                int[] top = st.peek();

                if (top[1] > tower[1]) {
                    receiveIdx = top[0];
                    break;
                }
                st.pop();
            }
            st.push(tower);
            answer[i] = receiveIdx;
        }
        return answer;
    }
}

 

결과

테스트 1 〉 통과 (1.55ms, 54.2MB)
테스트 2 〉 통과 (1.49ms, 52.1MB)
테스트 3 〉 통과 (1.65ms, 50.4MB)
테스트 4 〉 통과 (1.58ms, 52.2MB)
테스트 5 〉 통과 (1.56ms, 52.3MB)
테스트 6 〉 통과 (1.69ms, 50.8MB)
테스트 7 〉 통과 (1.55ms, 52.2MB)
테스트 8 〉 통과 (1.74ms, 50.3MB)

 

복기

단순하게 2중 for문으로 푸는 법 밖에 생각이 나지 않았다.
스택/큐 카테고리 문제라 이 자료구조들을 사용하여 푸는데 어려움을 겪어 다른 사람의 풀이를 참고했다.
현재 문제의 테스트에서는 두 풀이방법에서의 시간, 메모리 차이가 거의 없었다.
일부는 타워의 개수가 많을 때 스택을 사용한 풀이가 훨씬 효율적이라고 한다.
굳이 탐색 하지 않아도 되는 타워들을(낮은 타워)를 배제해서 그런가 싶다.
근데 체감상 잘 모르겠다.