•
문제 설명 : 조건에 맞는 인덱스 목록을 오름차순으로 반환한다.
•
풀이 방법
◦
탐색 범위 양끝을 지정한 후, key 존재를 확인하였다.
◦
가 만족하는 범위로 설정하였다.
•
시간복잡도 :
•
성공 코드
class Solution:
def findKDistantIndices(self, nums: List[int], key: int, k: int) -> List[int]:
result = []
for i in range(len(nums)):
# 탐색 범위 확인
start = 0 if i - k <= 0 else i - k
last = len(nums) - 1 if i + k >= len(nums) else i + k
# key 존재 확인
for j in range(start, last + 1):
if nums[j] == key:
result.append(i)
break
return result
Python
복사