编写一个函数,判断一个数是否为质数的函数