当前编程题:线性时间内寻找元素—线性时间内寻找元素(递归与分治)
【问题描述】每次都是优化选出一个元素为划分基准,在线性时间内寻找第i小元素。
【输入形式】在屏幕上输入若干整数,各数间都以一个空格分隔。再输入要寻找的元素是数组从小到大顺序中第几个位置。
【输出形式】数组从小到大顺序中要寻找的那个位置的元素。
【样例输入】
2 9 8 0 7 10 1 12 3 14 5 13 6 11 4
3
【样例输出】
2
【样例说明】
输入:15个整数,以空格分隔。要寻找第3小元素。
输出:2,表示第3小元素为2。
【评分标准】根据输入得到准确的输出。
思想:
算法导论上的伪代码:
代码:
import numpy as np
def Partition(L, p, r): # 划分数组,返回基准元素下标
x = L[r]
i = p-1
for j in range(p,r):
if L[j] <= x:
i = i+1
L[i], L[j] = L[j], L[i]
L[i+1], L[r] = L[r], L[i+1]
return i+1
def Select(L, p, r, i): # 在L[p]到L[r]中找第i小的数
cur = Partition(L, p, r)
k = cur-p+1
if i == k:
return L[cur]
elif i < k:
return Select(L, p, cur-1, i)
else:
return Select(L, cur+1, r, i-k)
def main():
L = np.array(list(map(int, input().split())))
n = len(L)-1
j = input()
j = int(j)
c = Select(L, 0, n, j) # 报错
print(c)
if __name__ == '__main__':
main()