Mercurial > public > think_complexity
view ch3ex3.py @ 20:0326803882ad
Finally completing Ch. 3, exercise 5: write a TreeMap that uses a red-black
tree.
author | Brian Neal <bgneal@gmail.com> |
---|---|
date | Thu, 27 Dec 2012 15:30:49 -0600 |
parents | 0f98bcb5bd3f |
children | cfb7f28678c7 |
line wrap: on
line source
"""Chaper 3.3, exercise 3. Write a function called bisection that takes a sorted list and a target value and returns the index of the value in the list if it's there, or None if it's not. """ def bisection(a, item): """Returns the index of item in the sorted list a if it exists or None if not. Finds the "rightmost" item if it exists. """ lo = 0 hi = len(a) while lo < hi: mid = (lo + hi) // 2 if item < a[mid]: hi = mid else: lo = mid + 1 n = lo - 1 return a[n] if n >= 0 and a[n] == item else None