53-1- 정렬 된 배열에서 숫자 찾기 -python

주제 : 정렬 된 배열에 숫자가 나타나는 횟수입니다. 입력은 정렬 된 배열과 숫자입니다.

def get_first_num(nums,k,start,end):
    if start>end:
        return -1
    mid = (end+start)//2
    if nums[mid]==k:
        if mid==0 or (mid>0 and nums[mid-1]!=k):
            return mid
        else:
            end = mid-1
    elif nums[mid]>k:
        end = mid-1
    else:
        start = mid+1

    return get_first_num(nums,k,start,end)

def get_end_num(nums,k,start,end):
    if start>end:
        return -1
    mid = (end + start) // 2
    if nums[mid]==k:
        if mid==len(nums)-1 or (mid<len(nums)-1 and nums[mid+1]!=k):
            return mid
        else:
            start = mid
    elif nums[mid]>k:
        end = mid-1
    else:
        start = mid+1
    return get_end_num(nums,k,start,end)

def get_num_k(nums,k):
    start = get_first_num(nums,k,0,len(nums)-1)
    end = get_end_num(nums,k,0,len(nums)-1)
    print(start,end)
    return end-start+1

　　참고 : 일반적인 해결책은 사전을 사용하여 각 숫자의 발생 횟수를 계산하거나 이분법을 사용하여 대상 숫자 중 하나를 찾은 다음 앞뒤로 검색하는 것입니다. 이 두 방법의 시간 복잡도는 O (n)입니다.

O (lgn) 방법은 두 개의 이진 검색을 사용하는 것입니다. 처음에는 대상 번호의 첫 번째 발생 위치를 찾고, 두 번째로 마지막 발생 위치를 찾고, 둘의 마지막 빼기는 다음과 같습니다. 대상 번호의 번호입니다.