你们都是TLE,只有我是MLE???
查看原帖
你们都是TLE,只有我是MLE???
156899
SakuraTao楼主2021/1/21 22:07

用的是分治,但是后两个MLE了,呜呜呜,是不是python必然MLE,到现在还没有一个用Python通过的,哭了

n,k = map(int,input().strip().split())
num = input().strip().split()
num = [int(i) for i in num]
def partition(seq):
    pi = seq[0]
    lo = [x for x in seq[1:] if x<=pi]
    hi = [x for x in seq[1:] if x>pi]
    return lo,pi,hi
def select(seq,k):
    lo,pi,hi = partition(seq)

    m = len(lo)
    if m == k:
        return pi
    elif m < k:
        return select(hi,k-m-1)
    else:
        return select(lo,k)
print(select(num,k))

2021/1/21 22:07
加载中...