프로그래머스/기초

[프로그래머스] Lv.0 /수열과 구간 쿼리 2 (파이썬/Python)

junslee 2025. 2. 17. 07:59

1. 문제 설명

2. 풀이 과정

answer에 정답을 넣기 위해 빈 문자열로 초기화한다.

for문으로 s,e,k에 queries의 원소를 받는다.

arr_i에 s<=i<=e 범위에 있있으며 k보다 큰 배열을 저장한다.

arr_i가 존재하면 min()으로 최소값을 answer에 추가하고,

존재하지 않으면 -1를 추가한다.

for문을 나오면 answer를 반환한다.

3. 코드

def solution(arr, queries):
    answer = []
    for s, e, k in queries:
        arr_i = [arr[i] for i in range(s, e+1) if arr[i] > k]
        if arr_i:
            answer.append(min(arr_i))
        else:
            answer.append(-1)
    return answer