[Python] 프로그래머스 수열과 구간 쿼리 2

문제 설명

정수 배열 arr와 2차원 정수 배열 queries이 주어집니다. 

queries의 원소는 각각 하나의 query를 나타내며, [s, e, k] 꼴입니다.
각 query마다 순서대로 s ≤ i ≤ e인 모든 i에 대해 k보다 크면서 가장 작은 arr[i]를 찾습니다.
각 쿼리의 순서에 맞게 답을 저장한 배열을 반환하는 solution 함수를 완성해 주세요.
단, 특정 쿼리의 답이 존재하지 않으면 -1을 저장합니다.

 

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

프로그래머스

제한 사항

1 ≤ arr의 길이 ≤ 1,000
0 ≤ arr의 원소 ≤ 1,000,000
1 ≤ queries의 길이 ≤ 1,000
0 ≤ s ≤ e < arr의 길이
0 ≤ k ≤ 1,000,000

예제

arr = [0, 1, 2, 4, 3]
queries = [[0, 4, 2],[0, 3, 2],[0, 2, 2]]
result = [3, 4, -1]

문제 풀이

def solution(arr, queries):
    answer = []
    for s, e, k in queries:
    	# 쿼리 범위에 해당하는 arr2 선언
        arr2 = arr[s:e+1]
        
        # 최소값 구하기
        min_value = float('inf')  
        for i in arr2:
            if k < i < min_value:  
                min_value = i
                
		# min_value가 변함이 없다면 -1      
        if min_value == float('inf'):  
            answer.append(-1)          
        else:
            answer.append(min_value)
    return answer

---

파이썬을 독학하시는 분들에게 도움이 되길 바라며,
혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.