[알고리즘 문제풀이] #4 탑[JAVA]

 

 

1. 문제 설명

 

 

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

 

 

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

 

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

 

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

 

 

 

2. 나의 코드

 

Stack을 사용하여 풀었다.

import java.util.*;

class Solution {
    public int[] solution(int[] heights) {
        Stack <integer> st = new Stack<integer>();
        int [] answer = new int[heights.length];
        int k=0;
        Arrays.fill(answer,0);

        for(int i =0 ; i < heights.length ;i++) {
            st.push(heights[i]);
        }

        for(int i = answer.length-1; i >= 0 ;i--){
            k = st.pop();

            for(int j = i; j >= 0; j--){
                if(k < heights[j]){
                    answer[i] = j+1;
                    break;
                }
            }
        }
        return answer;
    }
}

 

 

3. 다른 사람의 코드

 

 

for (int i=0; i < heights.length; i++){
            for (int j=i+1; j < heights.length; j++){
                if (heights[i] > heights[j]){
                    answer[j]=i+1;
                }
            }
        }

 

 

4. 평가

 

Stack을 사용하여 뒤에서 부터 순차적으로 비교연산을 할수 있었다. 

하지만, 앞에서 푸는 것이 더욱 쉽고 간단하다. 

다른 사람들의 코드를 보면, 문제에서 주어진 대로 풀지 않았다.  

앞에서부터 차례대로 풀었고,

배열의 앞에서 뒤로 갈때마다 answer 배열이 갱신되는 것을 볼 수 있다.