题目描述：编写一个函数，判断一个整数是否为素数，并在主函数中输入一个整数，输出"YES"表示是素数，"NO"表示不是素数。

输入格式：一行，一个整数。

输出格式：一行，输出"YES"或"NO"。

输入示例：

17

输出示例：

YES

难度：基础