在R中编写素数判断函数如何修复输入0返回NA的问题

• 素数的数学定义为大于1的自然数，除了1和自身外无法被其他数整除，0本身不符合素数的基本定义，本应返回FALSE
• 原函数输入n=0时，1:n会生成向量c(1, 0)，0 %% 0的运算结果为NaN，求和后得到NA，最终导致判断结果异常

在函数开头新增边界判断逻辑，所有小于等于1的输入直接返回FALSE，修改后的函数代码如下：

is_prime <- function(n){
  # 边界判断：小于等于1的数均不符合素数定义
  if(n <= 1){
    return(FALSE)
  }
  ifelse(sum(n %% (1:n)==0) == 2, TRUE, FALSE) 
}

修改后再运行原测试代码，输入0就会正常返回FALSE。

原函数判断因数时遍历到n存在冗余计算，实际只需遍历到sqrt(n)即可得到正确结果，优化后代码运行效率更高：

is_prime <- function(n){
  if(n <= 1){
    return(FALSE)
  }
  if(n == 2){
    return(TRUE)
  }
  ifelse(sum(n %% (2:floor(sqrt(n))) == 0) == 0, TRUE, FALSE)
}

内容的提问来源于stack exchange，提问作者Balina