one line of code at a time

[leetcode] 704. Binary Search 파이썬 코드 본문

leetcode

[leetcode] 704. Binary Search 파이썬 코드

oloc 2024. 8. 1. 10:32

정렬이 되어 있으면 binary search를 쓰는 것이 좋다. 

# binary search
nums = [-1,0,3,5,9,12]
target = 2

def bsearch(nums):
    lp, rp = 0, len(nums) - 1

    while lp <= rp:
        mid = (lp + rp) // 2

        if nums[mid] == target:
            return mid
        elif nums[mid] < target:
            lp = mid + 1
        else:
            rp = mid - 1

    return -1

print(bsearch(nums))