4_02 素数判定(自定义函数）

编写函数，参数是一个正整数n，如果它是素数，返回1，否则返回0。