Python3-PAT乙级1030 完美数列 (25 分)
给定一个正整数数列,和正整数 p,设这个数列中的最大值是 M,最小值是 m,如果 M≤mp,则称这个数列是完美数列。
现在给定参数 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
这道题目用了两种方法写,但是都没完全通过。
第一种方法:
先确定一个数,然后遍历排序后的正整数列表,符合条件数列长度+1,不符合退出循环并输出数列长度。(第4个测试点运行超时)
代码:
#PAT 1030
n, p = input().split()
p = int(p)
n = int(n)
num = sorted(map(int, input().split()))
Min = 0
Max = -1
count = 1
if n == 0:
print(0)
if n == 1:
print(int(num[0]))
else:
for i in range(n - 1):
Min = num[i]
for j in range(i + count, n, 1):
if Min * p < num[j]:
break
count += 1
print(count)
第二种方法:
分别从排序后的正整数列表的头和尾开始,取头取尾,取头去尾,去头取尾
如果符合条件,直接算出长度并输出。
如果不符合条件,两头往中间各取一个数,重复上述步骤。
(第5个测试点答案错误)
代码:
#PAT 1030
n, p = input().split()
p = int(p)
n = int(n)
num = sorted(map(int, input().split()))
Min = 0
Max = -1
while True:
if Max - 1 == Min :
print(1)
if num[Min] * p >= num[Max]:
print(n+Max- Min +1)
break
if num[Min+1] * p >= num[Max]:
print(n+Max- (Min + 1)+1)
break
if num[Min] * p >= num[Max - 1]:
print(n+(Max-1) - Min + 1)
break
Max -= 1
Min += 1