PAT 1030 完美数列 python (无超时)

1030 完美数列(25)(25 分)

给定一个正整数数列,和正整数p,设这个数列中的最大值是M,最小值是m,如果M <= m * p,则称这个数列是完美数列。

现在给定参数p和一些正整数,请你从中选择尽可能多的数构成一个完美数列。

输入格式:

输入第一行给出两个正整数N和p,其中N(<= 10^5^)是输入的正整数的个数,p(<= 10^9^)是给定的参数。第二行给出N个正整数,每个数不超过10^9^。

输出格式:

在一行中输出最多可以选择多少个数可以用它们组成一个完美数列。

输入样例:

10 8
2 3 20 4 5 1 6 7 8 9

输出样例:

8

作者: CAO, Peng
单位: PAT联盟
时间限制: 300ms
内存限制: 64MB
代码长度限制: 16KB

问题分析:
本题最初使用的循环嵌套,会出现运行超时
改进后对完美数列长度进行判断,过滤掉部分循环
同时,多次转制会造成运算超时,所以将转制放在最初input的位置

原代码:

x = [int(i) for i in input().split()]
n = input().split()
n = sorted(map(int, n))
rst = [1]
for i in range(int(x[0])):
    for j in range(int(x[0]) - 1, i, -1):
        if n[i] * int(x[1]) >= n[j]:
            rst.append(j - i + 1)
            break
print(max(rst))

改进代码:

x = [int(i) for i in input().split()]
n = input().split()
n = sorted(map(int, n))
m = 0
for i in range(x[0]):
    while m + i < x[0]:
        if n[i] * x[1] >= n[i + m]:
            m += 1
        else:
            break
print(m)

猜你喜欢

转载自blog.csdn.net/qq_42800234/article/details/81664053