[프로그래머스] 탑(42588) (java)

원본 문제 : https://programmers.co.kr/learn/courses/30/lessons/42588

문제 참고 : https://jhnyang.tistory.com/125

 

문제 설명

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

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

 

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

 

제한 사항

• heights는 길이 2 이상 100 이하인 정수 배열입니다.
• 모든 탑의 높이는 1 이상 100 이하입니다.
• 신호를 수신하는 탑이 없으면 0으로 표시합니다.

입출력 예

 

입출력 예 설명

 

입출력 예 #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.Arrays;
import java.util.Stack;

// https://programmers.co.kr/learn/courses/30/lessons/42588

public class LESSON42588 {

	public static void main(String[] args) {
		int[] heights = {3,9,9,3,5,7,2};
		
		System.out.println(Arrays.toString(solution(heights)));
		
	}
	
    public static int[] solution(int[] heights) {
    	//리턴되는 배열의 길이는 heights와 같음
        int[] answer = new int[heights.length];
        
        //Sender들을 저장하는 Stack을 생성
        Stack<Integer> sender_st = new Stack<Integer>();
        
        //sender의 길이는 배열의 길이와 같음
        int sender_len = heights.length;
        
        //stack에 sender를 차례대로 push();
        for(int height : heights) {
        	sender_st.add(height);
        }
        
        //sender와 receiver 변수 선언
        int sender, receiver;
        
        //sender를 받을 수 있는 receiver를 찾기 위한 루프
        for(int sender_idx = sender_len - 1; sender_idx > 0 ; sender_idx--) {
        	
        	//가장 마지막에 저장된 sender를 pop();
        	sender = sender_st.pop();
        	
        	//sender가 보내는 신호를 수신할 receiver들을 저장하는 stack 생성
        	Stack<Integer> receiver_st = new Stack<Integer>();
        	
        	//receiver의 길이는 sender_len-1만큼 길이
        	int receiver_len = sender_idx;
        	
        	//receiver_st에 receiver들을 하나씩 순서대로 저장
        	for(int i = 0 ; i < receiver_len; i++) {
        		receiver_st.add(heights[i]);
        	}
        	
        	//receiver_st에서 receiver를 하나씩 꺼내서 sender들과 비교할 루프
        	for(int receiver_idx = receiver_len - 1; receiver_idx >= 0 ; receiver_idx--) {
        		//receiver 하나를 pop();
        		receiver = receiver_st.pop();
        		
        		//sender보다 큰 receiver를 찾으면
        		if(sender<receiver) {
        			//answer배열의 자리 sender_idx에 찾은 receiver의 위치인 receiver_idx+1을 대입
        			answer[sender_idx] = receiver_idx+1;
        			//break로 루프를 탈출하여 다음 sender의 receiver를 찾음
        			break;
        		}
        	}
        }   
        return answer;
    }
}