L1-028 判断素数（Lua）

题目

本题的目标很简单，就是判断一个给定的正整数是否素数。

输入格式：
输入在第一行给出一个正整数N（≤ 10），随后N行，每行给出一个小于2
31
的需要判断的正整数。

输出格式：
对每个需要判断的正整数，如果它是素数，则在一行中输出Yes，否则输出No。

输入样例：
2
11
111
输出样例：
Yes
No

代码

function isPrime(x)
    if x == 0 or x == 1 then
        return false
    end
    
    local i = 2
    while i <= x / i do
        if x % i == 0 then
            return false
        end
        
        i = i + 1
    end
    
    return true
end

local n = io.read()

for i = 1, n do
    local num = tonumber(io.read())

    if isPrime(num) then
        print("Yes")
    else
        print("No")
    end
end