Discussion on Find the Median Challenge

6 years ago+ 3 comments

i know this problem can be easily solved by sorting the array and returning the middle elemrnt. But I wanted to challenge myself and reduce the run time from O(nlogn) to O(n). Quick Select shown below implemented in a pythonic way has that run time. Hope it helps!

def partition(l, start, end):
    pivot = l[start]
    bound = start + 1
    for i in range(start+1, end+1):
        if l[i] < pivot:
            l[i], l[bound] = l[bound], l[i]
            bound += 1
            
    bound -= 1
    l[start], l[bound] = l[bound], l[start]
    return bound

def find_median(l, start, end, k):
    if(start == end):
        return l[end]
    
    cur = partition(l, start, end)
    if cur == k:
        return l[k]
    elif cur > k:
        return find_median(l, start, cur - 1, k)
    else:
        return find_median(l, cur + 1, end, k)
        
n = int(input())
l = list(map(int, input().split()))

k = n // 2
    
print(find_median(l,0,n-1,k))