프로그래머스 탑 Java 풀이

출처

https://programmers.co.kr/learn/courses/30/lessons/42588(스택/큐 카테고리에 있었으나 없어졌습니다.)

문제 설명

수평 직선에 탑 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번째 탑이 수신합니다.

기존 풀이

import java.util.Stack;

class Solution {
    public int[] solution(int[] heights) {
        int[] answer = new int[heights.length];
        Stack<Integer> stk = new Stack<>();

        for(int i=0;i<heights.length;i++) {
            stk.push(heights[i]); //heights 배열의 값을 스택에 넣는다.
        }

        while(!stk.isEmpty()) {
            int tmp=stk.pop();

            for(int i=stk.size();i>=0;i--) { // height 배열의 오른쪽부터 왼쪽으로 탐색한다.
                if(tmp<heights[i]) { // tmp(스택에서 pop한 값)이 배열의 값보다 작으면
                    answer[stk.size()]=i+1; // 배열 index + 1한 값을 답안 배열에 할당한다.
                    break;
                }
            }
        }
        return answer;
    }
}

출처: https://saltae.tistory.com/5

다른 사람의 풀이(HamYoungChan님의 풀이)

import java.util.Stack;

public class Top {
	public int[] solution(int[] heights) {
        Stack<Tower> st = new Stack<>();
        int[] answer = new int[heights.length];

        for (int i = 0; i < heights.length; i++) {
            Tower tower = new Tower(i + 1, heights[i]);
            int receiveIdx = 0;

            while (!st.isEmpty()) {
                Tower top = st.peek(); // peek을 통해 스택의 요소를 삭제하지 않고 리턴한다.

                if (top.height > tower.height) {
                    receiveIdx = top.idx; // top(왼쪽 탑)의 높이가 해당 탑보다 높으면 그 인덱스를 할당한다.
                    break;
                }

                st.pop(); // 여기서 pop을 하기 때문에 왼쪽 탑에는 지나간 인덱스의 탑보다 높은 탑만 남게 된다.
            }

            st.push(tower);
            answer[i] = receiveIdx;
        }

        return answer;
    }

    class Tower {
        int idx;
        int height;

        public Tower(int idx, int height) {
            this.idx = idx;
            this.height = height;
        }

        @Override
        public String toString() {
            return "idx : " + idx + " height : " + height;
        }
    }
}

스택 내 마지막 요소를 삭제하고 리턴하는 pop 메소드와 달리 삭제하지 않고 리턴하는 peek 메소드를 써서

푼 풀이이다.

* 참고 : https://docs.oracle.com/javase/7/docs/api/java/util/Stack.html