一个数字中素数的和 - Lua

2014-12-3 18:18:35
收藏：0
阅读：68
评论：1

我正在尝试计算给定数字中素数的总和。例如，对于数字123456，结果将为10，因为2+3+5 = 10。

我尝试用Lua编写代码来实现这一目标，但遇到了一些问题。

首先，这是代码：

    function isPrime(num)
if(num == 1 or (num ~= 2 and num%2 == 0)) then
  return false
end
for i=3, math.sqrt(num), 2 do
  if(num%i == 0) then
    return false
  end
end
return true
end

function sumOfPrimes(num)
local sum = 0
for digit in string.gmatch(num,"%d") do
  local prime = isPrime(digit)
  if(isPrime(digit)) then
    sum = sum + digit
  end
  print(digit)
end
return sum
end

function main()
print(sumOfPrimes(123456))
end
main()

它返回9而不是10。另一件我注意到的事情是它还将1添加到总和中，但1不是素数。问题是什么？

点赞
用户1058509
用户1058509

string.gmatch 返回一个字符串，你需要在计算之前将其转换为数字。

顺便说一句，你在循环中进行了两次素数验证。

这是一个修复后的版本（按预期返回10）：

...

function sumOfPrimes(num)
    local sum = 0
    for digit in string.gmatch(num, "%d") do
        digit = tonumber(digit) -- 需要转换
        local prime_digit = isPrime(digit)
        if prime_digit then
            sum = sum + digit
        end
    end
    return sum
end
2014-12-03 18:43:52

作者:

用户3889576

用户3889576
推荐文章