파이썬 문제: 4/7

프로그래머스 Level2 : 주식가격

초 단위로 기록된 주식가격이 담긴 배열 prices가 매개변수로 주어질 때, 가격이 떨어지지 않은 기간은 몇 초인지를 return 하도록 solution 함수를 완성하세요.

제한사항

• prices의 각 가격은 1 이상 10,000 이하인 자연수입니다.
• prices의 길이는 2 이상 100,000 이하입니다.

입출력 예

prices	return
[1, 2, 3, 2, 3]	[4, 3, 1, 1, 0]

입출력 예 설명

• 1초 시점의 ₩1은 끝까지 가격이 떨어지지 않았습니다.
• 2초 시점의 ₩2은 끝까지 가격이 떨어지지 않았습니다.
• 3초 시점의 ₩3은 1초뒤에 가격이 떨어집니다. 따라서 1초간 가격이 떨어지지 않은 것으로 봅니다.
• 4초 시점의 ₩2은 1초간 가격이 떨어지지 않았습니다.
• 5초 시점의 ₩3은 0초간 가격이 떨어지지 않았습니다.

※ 공지 - 2019년 2월 28일 지문이 리뉴얼되었습니다.

해설

처음엔 enumerate를 사용했는데 효율성에서 자꾸 탈락했다. 그래서 이를 range로 바꿨는데 통과했다.

def solution(prices):
    answer = []           
    for idx in range(len(prices)):
        answer.append(0)
        for i in range(idx+1, len(prices)): #값이 하락하지 않으면 유지일수 증가
            answer[-1] += 1
            if prices[idx] > prices[i]: #값이 하락하면 끝
                break
    return answer

다른 사람 풀이를 보니 원래 의도한대로 stack을 사용한 풀이가 있었다. 근데 굳이 stack을 써야하나 싶었는데, 테스트 3에서 위 풀이는 209ms, 아래 풀이는 44ms가 나온걸 봐서 똑같은 2중 for문이라도 효율성이 다르다는 걸 알 수 있었다.

def solution(prices):
    stack = [] #값 하락이 발견되지 않은 인덱스를 저장
    answer = [0] * len(prices)
    for i in range(len(prices)):
        if stack != []:
            while stack != [] and stack[-1][1] > prices[i]:
                #스택 값들보다 현재값이 작으면 값이 하락되었다는 의미
                #하락되지 않을떄까지 stack을 비워서 스택이 비었거나, 값 하락이 없으면 끝냄
                past, _ = stack.pop()
                answer[past] = i - past
        stack.append([i, prices[i]]) #stack이 비었으면 인덱스와 값을 넣어줌
    for i, s in stack:
        answer[i] = len(prices) - 1 - i
    return answer

프로그래머스 Level2 : 탑

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

해설

신호를 왼쪽으로 쏘기때문에, heights를 뒤에서 부터 접근했다.

def solution(heights):
    for i in range(len(heights)):
        for j in range(i+1, len(heights)):
            if heights[-i-1] < heights[-j-1]:
                #맨 뒤부터 접근해 탑 높이가 높으면 신호 수신
                heights[-i-1] = len(heights)-j
                #다른 탑이 신호를 받으면 그 탑의 높이는 필요 없으므로 height를 답으로 바꿈
                break
            if j == len(heights) - 1:#수신 받는 탑이 없는 경우
                heights[-i-1] = 0
    return [0] + heights[1:]
	#제일 첫번째 탑은 수신 받을 탑이 없으므로 0, answer는 현재 맨뒤부터 접근해서 역순이므로 복구

근데 생각해보면 range를 그냥 거꾸로 불러오면 더 편하다. 그런 풀이가 있더라.

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

근데 stack 문제라 그런지 stack을 이용한 풀이도 있다. 내가 stack 개념을 잘 몰라서 그럴 수도 있는데 위 풀이보다는 덜 직관적인 듯. 그래도 생각해보면 iteration이 위 풀이보다 작다.

def solution(heights):
    answer = [0]*len(heights)
    stack = []
    for i in reversed(range(len(heights))):
        while stack and stack[-1][1] < heights[i]:
            #stack이 비었거나, 가장 위의 값이 현재 탑 높이보다 크면 끝냄
            #가장 위 값이 현재 탑 높이보다 작다는건, 현재 탑이 신호를 수신했다는 의미
            idx, height = stack.pop()
            answer[idx] = i+1
        stack.append((i, heights[i]))
    return answer