素数判定

いろいろな場所で素数判定をおこなうコードはあるんだろうけど、自分の学習用にR言語で作成したものを載せます。

is.prime

is.prime <- function(num){
    # if num is integer
    if (num != as.integer(num)) return("Input is not integer!")
    # if num is less than 2
    if (num < 2){
     return("Input is wrong!")
    } 
    # if num is 2 or 3
    else if (num == 2 || num == 3) {
        return(TRUE)
    } else {

        for (i in 2:ceiling(num/2)){

            if ((num %% i) == 0) {

                return(FALSE)

            }

        }

     return(TRUE)
    }
}

この関数Primeは引数に２よりも大きい自然数をとります。

上記のis.prime関数を利用して、指定した範囲の素数をベクトルで戻す関数も作ってみました。

getPrime


getPrime <- function(max, min = 2){
    if (max != as.integer(max) || max < 2) return("Maximum value is wrong!")

    else if (min != as.integer(min)|| min < 2) return("Minimum value is wrong!")
    else {
    res <- NULL
    for (i in min:max){
        if(is.prime(i) == TRUE) res <- append(res,i)

    }
    return(res)
    }
}