检查数字是否为素数

Question

输入是一个数字
如果数字是质数，则输出为 YES，否则输出为 NO

例一：
输入
97
输出
是

例子2：
输入
90
输出
否

Function PrimeOrNot(integer num) returns nothing 
   integer i
   integer k 
   
   for i = 2; i < num; i = i + 1
      if num % i == 0
         k = k + 1
   if k == 0
      Put "YES" to output 
   else
      Put "NO" to output 
      
Function Main() returns nothing 
   integer inputNum 
   
   inputNum = Get next input 
   
   PrimeOrNot(inputNum)

我的代码通过了除 1 以外的所有测试。不确定他们用什么测试了我的代码，但我似乎无法在我的程序中找到任何错误。任何帮助，将不胜感激。谢谢