分治法——查找最大最小元素(Python)

# 基本子算法(子问题规模小于等于 2 时)
def get_max(max_list):
    return max(max_list) 

def get_min(min_list):
    return min(min_list)

# 分治法 最小元素
def solve(init_list):
    n = len(init_list)
    if n <= 2: # 若问题规模小于等于 2,最终解决
        return get_min(init_list)

    # 分解(子问题规模为 2,最后一个可能为 1)
    temp_list=(init_list[i:i+2] for i in range(0, n, 2))
    
    # 分治,合并
    min_list = list(map(get_min, temp_list))
    # 递归(树)
    return solve(min_list)

    
        
# 分治法 最大元素
def solve2(init_list):
    n = len(init_list)
    if n <= 2: # 若问题规模小于等于 2,解决
        return get_max(init_list)

    # 分解(子问题规模为 n/2)
    left_list, right_list = init_list[:n//2], init_list[n//2:]
    
    # 递归(树),分治
    left_max, right_max = solve2(left_list), solve2(right_list)
    
    # 合并
    return get_max([left_max, right_max])


if __name__ == "__main__":
    # 测试数据
    
    test_list = [13,4,11,5,67,7,9,43,24,2]
    # 求最小最大值
    print("最小元素:"+ str(solve(test_list)))  # 2
    print("最大元素:"+ str(solve2(test_list))) # 67

猜你喜欢

转载自blog.csdn.net/yummy_lym/article/details/79595581
今日推荐