[프로그래머스] 코딩테스트 풀이 - 탑

2019-10-01

문제 설명

수평 직선에 탑 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	각 탑이 쏜 신호를 어느 탑에서 받았는지 기록한 배열

문제를 풀기위해 주어진 데이터

# 문제 해결에 필요한 데이터

1. 탑의 길이

# 제한조건

1. 신호를 수신하는 탑이 없으면 0으로 표시

간단 알고리즘

1 2	1. 0번째 탑은 항상 수신하지 못한다. 2. 현재 탑의 왼쪽이 현재 탑의 높이보다 높은지 체크하고 높으면 배열에 넣는다.

나의 풀이

사용언어 : Python 3

간단하게 2중 for문을 사용해서 풀어주었다.
이번 예제를 통해 for else문을 알게되었다.

def solution(heights):
    answer = []

    for i, num in enumerate(heights):
        for j in range(i, -1, -1):
            if heights[j] > num:
                answer.append(j + 1)
                break
        else:
            answer.append(0)

    return answer

다른 풀이

1. 2중 for문

0을 먼저 선언해준뒤에 비교했다는 사실을 빼면 차이 없음.

def solution(h):
    ans = [0] * len(h)
    for i in range(len(h)-1, 0, -1):
        for j in range(i-1, -1, -1):
            if h[i] < h[j]:
                ans[i] = j+1
                break
    return ans

2. 스택을 사용한 방법

문제 의도가 스택인것같아 한번 더 봤다.
기본적으로는 비슷하지만 pop 을 이용해서 공간을 덜 먹는다는게 장점인듯

스택을 만들어주고 LIFO 법칙에 따라 가장 마지막 배열을 현재의 탑으로 잡고 배열에서 없앤다.

def solution(heights):
    responses = []
    while heights:
        top = heights.pop()

현재의 탑의 길이보다 -1부터 내려가면서 탐색한다. 있으면 answer에 append한다.

    for i in range(len(heights) - 1, -1, -1):
        if top < heights[i]:
            responses.append(i + 1)
            break

    else:
        responses.append(0)

return responses[::-1]

Taeyeon's blog

Algorithm프로그래머스