2015年2月13日星期五

用regular expression check prime

好神奇,竟然可以用一行regular expression就可以check prime number

import sys
import re

def is_prime(p):
    return re.search(r"^1?$|^(11+?)\1+$", '1' * p) == None

if __name__ == "__main__":
    print(is_prime(int(sys.argv[1])))

沒有留言:

發佈留言