写一判素数的函数，在主函数中输入一个整数，调用该函数进行判断并输出结果。