码学堂 | 一本通例6.7 素数

来源 : 信息学奥赛一本通

描述

写一个判断素数的函数，输入一个数，判断它是否是素数，是输出 yes，不是输出no。

输入

一个整数

输出

如果是素数输出yes，否则输出no

样例输入 1

样例输出 1

yes

样例输入 2

样例输出 2

no