我做了以下程序,要求用户输入上限,并计算每个完美平方的上限打印&。不过,我认为我的is_perfect_square
函数效率不高,因为当上限在数千或更多时,计算完美平方需要很长时间。我想知道如何让我的程序更高效,我认为使用数学模块来使用sqrt
可以工作,但我不是数学家,所以请求帮助。 我的计划是:如何有效地枚举所有完美的正方形?
"""Print all the perfect squares from zero up to a given maximum."""
import math
def read_bound():
"""Reads the upper bound from the standard input (keyboard).
If the user enters something that is not a positive integer
the function issues an error message and retries
repeatedly"""
upper_bound = None
while upper_bound is None:
line = input("Enter the upper bound: ")
if line.isnumeric() and int(line) >= 0:
upper_bound = int(line)
return upper_bound
else:
print("You must enter a positive number.")
def is_perfect_square(num):
"""Return true if and only if num is a perfect square"""
for candidate in range(1, num):
if candidate * candidate == num:
return True
def print_squares(upper_bound, squares):
"""Print a given list of all the squares up to a given upper bound"""
print("The perfect squares up to {} are: ". format(upper_bound))
for square in squares:
print(square, end=' ')
def main():
"""Calling the functions"""
upper_bound = read_bound()
squares = []
for num in range(2, upper_bound + 1):
if is_perfect_square(num):
squares.append(num)
print_squares(upper_bound, squares)
main()
这样做只能在'O(sqrt(upper-bound))'可实现时减少运行时到'O(upper_bound)'。 – Henrik