[Java] 자바 알고리즘, 탑(TOP) 문제 프로그래머스 level2

탑

문제 설명

수평 직선에 탑 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
[1,2,3] 번째 탑이 쏜 신호는 아무도 수신하지 않습니다.
[4,5,6] 번째 탑이 쏜 신호는 3번째 탑이 수신합니다.
[7] 번째 탑이 쏜 신호는 6번째 탑이 수신합니다.

입출력 예 #3
[1,2,4,5] 번째 탑이 쏜 신호는 아무도 수신하지 않습니다.
[3] 번째 탑이 쏜 신호는 2번째 탑이 수신합니다.
[6] 번째 탑이 쏜 신호는 5번째 탑이 수신합니다.
[7] 번째 탑이 쏜 신호는 6번째 탑이 수신합니다.

 

 

나의 풀이

class Solution {
    public int[] solution(int[] heights) {
        int[] answer = new int[heights.length];
        
        for(int i =0; i<heights.length; i++) {
        	int receptTop = getReceptTop(heights, i);
        	answer[i]= receptTop;
        }
        return answer;
    }

	private int getReceptTop(int[] heights, int nowTop) {
		int top= 0;
		int nowTopHeight = heights[nowTop];
		for(int i =nowTop; i>=0; i--) {
			if(heights[i] > nowTopHeight)
				return i+1;
		}
		
		return top;
	}
}

- 사실상 중요한 건 getRecepTop 메소드인 것 같습니다. 이건 사실 스택/큐 문제였는데요. 그냥 일반적인 풀이방법으로 풀었습니다. 스택구조로 어떻게 접근해야될지 모르겠네요ㅠㅠㅠ. 단순히 탑에서 한칸씩 감소시키면서 이전 탑과의 높이를 비교하였습니다. 중요한건 탑의 높이를 반환하는 것이 아닌, 탑의 인덱스(실제로는 인덱스+1 값이겠네요.)에 해당하는 순서를 반환하는 것입니다.

 

다른사람의 풀이

import java.awt.Point;
import java.util.Stack;

class Solution {
    public int[] solution(int[] heights) {
         int[] answer = new int[heights.length];

        Stack<Point> st = new Stack<Point>();

        for(int i=heights.length-1; i>=0; i--) {
            for(int j=i-1; j>=0; j--) {
                if(heights[i] < heights[j]) {
                    st.push(new Point(i+1, j+1));
                    break;
                } else if(j==0) {
                    st.push(new Point(i+1, 0));
                }
            }
        }

        int size = st.size();
        answer[0] = 0;
        for(int i=1; i<=size; i++) {
            answer[i] = st.pop().y;
        }

        return answer;
    }
}

- 스택으로 푼 풀이입니다. Point 객체로 몇번째 탑이 보낸 송신신호를 몇번째 탑이 받았는지의 정보를 가지네요. 그걸 스택에 담습니다. 시작점이 제일 오른쪽 탑에서부터 시작하기 때문에(왜냐하면 for문 구조가 -- 구조로 되어있기 때문) 스택을 pop할때는 1번째 탑부터 결과가 answer 배열에 담기게 되네요.