问题 B: 分金块
题目描述
有一个老板有一袋金块。
每个月将有两名雇员会因其优异的表现分别被奖励一个金块。
按规矩,排名第一的雇员将得到袋中最重的金块,
排名第二的雇员将得到袋中最轻的金块。
根据这种方式,除非有新的金块加入袋中,
否则第一名雇员所得到的金块总是比第二名雇员所得到的金块重。
如果有新的金块周期性的加入袋中,则每个月都必须找出最轻和最重的金块。
假设有一台比较重量的仪器,我们希望用最少的比较次数找出最轻和最重的金块。
本题要求用二分算法完成。
输入
输入数据有多行,第一行为金块个数n(2<=n<=5000000),接下来n行是n个金块的重量。
输出
输出最重的金块和最轻的金块,用空格隔开。
样例输入
8
10
8
2
4
5
3
9
1
样例输出
10 1
解答(分治算法):
# 二分法求列表ls在[lt,rt]内的最大值和最小值
def max_min(ls, lt, rt, res):
# 1.[lt,rt]区间的元素小于2个,直接求解
if rt - lt <= 2:
res[0] = max(ls[lt], ls[rt]) # 最大值
res[1] = min(ls[lt], ls[rt]) # 最小值
return
# 2.[lt,rt]元素个数大于2,二分
mid = (lt + rt) // 2
lres = [0] * 2
rres = [0] * 2
# 2.1左半区间[lt,mid]递归求解
max_min(ls, lt, mid, lres)
# 2.2右半区间[mid+1,rt]递归求解
max_min(ls, mid + 1, rt, rres)
# 3.合并
res[0] = max(lres[0], rres[0])
res[1] = min(lres[1], rres[1])
n = int(input())
ls = []
for i in range(n):
x = int(input())
ls.append(x)
res = [0] * 2
max_min(ls, 0, len(ls) - 1, res)
print(res[0], res[1])
答案不唯一,必定有更加优化的解法欢迎分享