用python写算法——查找笔记（二）

查找

定义：在一些数据元素中，通过一定的方法找出与给定关键字相同的数据元素的过程。
1、列表查找：从列表中查找指定元素

• 输入：列表、待查找元素
• 输出：元素下标（未找到元素时一般返回None或-1）

• 内置列表查找函数：index()

顺序查找

顺序查找：也叫线性查找，从列表第一个元素开始，顺序进行搜索，直至找到元素或搜索到列表最后一个元素为止。
时间复杂度：O(n)

def linear_search(data_set, value):
    for i in range(range(data_set)):
        if data_set[i] == value:
            return i
    return None

二分查找

前提是已经排好序的。
时间复杂度：O(logn)

def binary_search(li, val):
    left = 0
    right = len(li)-1
    while left <= right:   # 候选区有值
        mid = (left+right)//2
        if li[mid] == val:
            return mid
        elif li[mid] > val:   # 带查找的值在mid右侧
            right = mid - 1
        else:                 # li[mid] < val 带查找的值在mid右侧
            left = mid + 1
    else:
        return None
li = [1,2,3,4,5,6,7,8,9]
print(binary_search(li,3))

返回的是：下标2

比较顺序查找与二分查找所用的时间

from cal_time import *

@cal_time
def linear_search(li, val):
    for ind, v in enumerate(li):
        if v == val:
            return ind
    else:
        return None


@cal_time
def binary_search(li, val):
    left = 0
    right = len(li)-1
    while left <= right:   # 候选区有值
        mid = (left+right)//2
        if li[mid] == val:
            return mid
        elif li[mid] > val:   # 带查找的值在mid右侧
            right = mid - 1
        else:                 # li[mid] < val 带查找的值在mid右侧
            left = mid + 1
    else:
        return None
li = list(range(1000000))
linear_search(li, 38900)
binary_search(li, 38900)

结果为：

其中cal_time为装饰器

import time

def cal_time(func):
    def wrapper(*args, **kwargs):
        t1 = time.time()
        result = func(*args, **kwargs)
        t2 = time.time()
        print("%s running time: %s secs." % (func.__name__, t2 - t1))
        return result
    return wrapper