Python/searches/interpolation_search.py

141 lines
4.5 KiB
Python
Raw Normal View History

2017-10-15 16:11:10 +00:00
"""
This is pure python implementation of interpolation search algorithm
"""
2017-10-15 16:11:10 +00:00
def interpolation_search(sorted_collection, item):
"""Pure implementation of interpolation search algorithm in Python
Be careful collection must be ascending sorted, otherwise result will be
2017-10-15 16:11:10 +00:00
unpredictable
:param sorted_collection: some ascending sorted collection with comparable items
2017-10-15 16:11:10 +00:00
:param item: item value to search
:return: index of found item or None if item is not found
"""
left = 0
right = len(sorted_collection) - 1
while left <= right:
2019-10-05 05:14:13 +00:00
# avoid devided by 0 during interpolation
if sorted_collection[left] == sorted_collection[right]:
if sorted_collection[left] == item:
return left
else:
return None
2019-10-05 05:14:13 +00:00
point = left + ((item - sorted_collection[left]) * (right - left)) // (
sorted_collection[right] - sorted_collection[left]
)
2019-10-05 05:14:13 +00:00
# out of range check
if point < 0 or point >= len(sorted_collection):
2017-10-15 16:11:10 +00:00
return None
current_item = sorted_collection[point]
if current_item == item:
return point
else:
2019-10-05 05:14:13 +00:00
if point < left:
right = left
left = point
2019-10-05 05:14:13 +00:00
elif point > right:
left = right
right = point
else:
if item < current_item:
right = point - 1
else:
left = point + 1
2017-10-15 16:11:10 +00:00
return None
2019-10-05 05:14:13 +00:00
2017-10-15 16:11:10 +00:00
def interpolation_search_by_recursion(sorted_collection, item, left, right):
"""Pure implementation of interpolation search algorithm in Python by recursion
Be careful collection must be ascending sorted, otherwise result will be
2017-10-15 16:11:10 +00:00
unpredictable
First recursion should be started with left=0 and right=(len(sorted_collection)-1)
:param sorted_collection: some ascending sorted collection with comparable items
2017-10-15 16:11:10 +00:00
:param item: item value to search
:return: index of found item or None if item is not found
"""
2019-10-05 05:14:13 +00:00
# avoid devided by 0 during interpolation
if sorted_collection[left] == sorted_collection[right]:
if sorted_collection[left] == item:
return left
else:
return None
2019-10-05 05:14:13 +00:00
point = left + ((item - sorted_collection[left]) * (right - left)) // (
sorted_collection[right] - sorted_collection[left]
)
2019-10-05 05:14:13 +00:00
# out of range check
if point < 0 or point >= len(sorted_collection):
2017-10-15 16:11:10 +00:00
return None
if sorted_collection[point] == item:
return point
2019-10-05 05:14:13 +00:00
elif point < left:
return interpolation_search_by_recursion(sorted_collection, item, point, left)
2019-10-05 05:14:13 +00:00
elif point > right:
return interpolation_search_by_recursion(sorted_collection, item, right, left)
2017-10-15 16:11:10 +00:00
else:
if sorted_collection[point] > item:
2019-10-05 05:14:13 +00:00
return interpolation_search_by_recursion(
sorted_collection, item, left, point - 1
)
else:
2019-10-05 05:14:13 +00:00
return interpolation_search_by_recursion(
sorted_collection, item, point + 1, right
)
2017-10-15 16:11:10 +00:00
def __assert_sorted(collection):
"""Check if collection is ascending sorted, if not - raises :py:class:`ValueError`
2017-10-15 16:11:10 +00:00
:param collection: collection
:return: True if collection is ascending sorted
:raise: :py:class:`ValueError` if collection is not ascending sorted
2017-10-15 16:11:10 +00:00
Examples:
>>> __assert_sorted([0, 1, 2, 4])
True
>>> __assert_sorted([10, -1, 5])
Traceback (most recent call last):
...
ValueError: Collection must be ascending sorted
2017-10-15 16:11:10 +00:00
"""
if collection != sorted(collection):
2019-10-05 05:14:13 +00:00
raise ValueError("Collection must be ascending sorted")
2017-10-15 16:11:10 +00:00
return True
2019-10-05 05:14:13 +00:00
if __name__ == "__main__":
2017-10-15 16:11:10 +00:00
import sys
"""
user_input = input('Enter numbers separated by comma:\n').strip()
2017-10-15 16:11:10 +00:00
collection = [int(item) for item in user_input.split(',')]
try:
__assert_sorted(collection)
except ValueError:
sys.exit('Sequence must be ascending sorted to apply interpolation search')
2017-10-15 16:11:10 +00:00
target_input = input('Enter a single number to be found in the list:\n')
2017-10-15 16:11:10 +00:00
target = int(target_input)
"""
debug = 0
if debug == 1:
2019-10-05 05:14:13 +00:00
collection = [10, 30, 40, 45, 50, 66, 77, 93]
try:
__assert_sorted(collection)
except ValueError:
2019-10-05 05:14:13 +00:00
sys.exit("Sequence must be ascending sorted to apply interpolation search")
target = 67
2017-10-15 16:11:10 +00:00
result = interpolation_search(collection, target)
if result is not None:
2019-10-05 05:14:13 +00:00
print("{} found at positions: {}".format(target, result))
2017-10-15 16:11:10 +00:00
else:
2019-10-05 05:14:13 +00:00
print("Not found")