简答题

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