编写函数，判断一个数字是否为素数，是则返回字符串YES，否则返回字符串NO。