2004/12/27

年明けの嵐が怖いので、大好き素数を求めて気を落ち着けるしか。

#!/usr/local/bin/python

def get_prime_list(num):
known_prime = [2]
current_integer = 3
while len(known_prime) < num:
for indy in known_prime:
if current_integer % indy == 0:
current_integer = current_integer + 2
break
else:
known_prime.append(current_integer)
current_integer = current_integer + 2
return known_prime

import sys
integer = int(sys.argv[1])
print integer,"th prime is",get_prime_list(integer)[integer-1]

current_integer = current_integer + 2 の部分がちょっと気が利いているような気がしていたんだけど、実行速度は +1 でもほとんどかわらないんじゃん。

0 件のコメント:

コメントを投稿