[Algorithm] [Python] Programmers - 탑

Programmers - 탑

---

문제 설명

---

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

예를 들어 높이가 6, 9, 5, 7, 4인 다섯 탑이 왼쪽으로 동시에 레이저 신호를 발사합니다. 그러면, 탑은 다음과 같이 신호를 주고받습니다.

높이가 4인 다섯 번째 탑에서 발사한 신호는 높이가 7인 네 번째 탑이 수신하고, 높이가 7인 네 번째 탑의 신호는 높이가 9인 두 번째 탑이, 높이가 5인 세 번째 탑의 신호도 높이가 9인 두 번째 탑이 수신합니다.

높이가 9인 두 번째 탑과 높이가 6인 첫 번째 탑이 보낸 레이저 신호는 어떤 탑에서도 수신할 수 없습니다.

제한 사항

---

• 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]

제출 코드

---

Python

def solution(heights):
    answer = [0] * len(heights)
    for i in range(len(heights)-1, -1, -1):          # heights 원소 거꾸로 순회
        for j in range(i-1, -1, -1):                 # 위 원소보다 한 칸 앞서있는 원소부터 거꾸로 순회
            if heights[j] > heights[i]:             # 만일 더 크면
                answer[i] = j+1
                break
    return answer

 

Swift

func solution(_ heights:[Int]) -> [Int] {
    var arr = heights
    var answer = Array(repeating: 0, count: arr.count)

    while arr.count != 0 {
        let target = arr.popLast()
        for i in stride(from: arr.count-1, through: 0, by: -1) {
            if arr[i] > target! {
                answer[arr.count] = i+1
                break
            }
        }
    }

    return answer
}

Swift 풀이 또한 알고리즘은 동일하다.

 

코드 설명

---

우선 정답을 담아 반환할 리스트를 생성한다.

이후, for문을 통해 뒤에서 앞 쪽으로 접근한다.

for i in range(초기, 조건, 증감): 의 의미는 초기값 부터 조건값 까지 증감연산을 진행하며 반복하겠다는 의미이다.

또한, 조건값은 포함되지 않기때문에 이를 고려하여 조건값을 기재하여야 한다.

cf) for i in range(0, 10)은 0~9까지 인덱싱을 진행한다.

따라서 for i in range(len(heights) -1, -1, -1): 구문의 의미는

heights 배열의 길이 -1 값 부터 -1까지 -1씩 연산을 진행하며 인덱싱 하겠다는 의미이다.

이를 통해 배열의 끝 원소부터 첫 원소까지 접근이 가능하다.

이와 같은 개념으로 이중 for문을 구성한다.

중첩 for문은 outer for문에서 인덱싱 중인 원소 i값 보다 1 작은 값부터 -1까지 -1연산을 진행하며 반복한다.

이를 반복하며 만일 heights[j] 값이 heights[i] 값 보다 크면 이를 정답 배열에 추가한다.

다른 사람의 풀이

---

def solution(heights):
    answer = [0] * len(heights)
    while heights:
        right = heights.pop()
        for idx in range(len(heights)-1, -1, -1):
            if heights[idx] > right:
                answer[len(heights)] = idx + 1
                break
    return answer

위 풀이는 stack 자료구조의 LIFO 성질을 이용해 필자의 outer for문을 대체하였다.

heights 배열에서 pop 을 통해 제일 끝의 원소를 빼오고 크기 비교를 통해 정답 배열에 값을 추가하였다.