编写一函数prime实现判断一个正整数是否为素数。要求在主函数main中输入一个正整数，并输出是否素数的信息。      例如：若输入11，则输出YES；若输入10，则输出NO。

正确答案: