Algorithm

[프로그래머스] 수열과 구간 쿼리 2 - 파이썬

potato_pizza 2024. 3. 13. 15:45
728x90

Programmers

수열과 구간 쿼리 2

https://school.programmers.co.kr/learn/courses/30/lessons/181923

[프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr](https://school.programmers.co.kr/learn/courses/30/lessons/181923)

문제

코드

  • 풀이
def solution(arr, queries):
    answer = []

    for s, e, k in queries:
        max_val = -1  
        for i in range(s, e + 1):
            if arr[i] > k and (max_val == -1 or arr[i] < max_val):
                max_val = arr[i]
        answer.append(max_val)

    return answer
  • 다른 풀이
def solution(arr, queries):
    answer = []
    for s, e, k in queries:
        tmp = []
        for x in arr[s:e+1]:
            if x > k:
                tmp.append(x)
        answer.append(-1 if not tmp else min(tmp))
    return answer

Reference

728x90
반응형