ruby - Ruby中的素数测试

Question

我有一个数组a = [1,2,3,4,5]。我想测试哪些数字是素数并想产生输出{1=>false, 2=>true, 3=>true, 4=>false, 5=>true}。

任何一个班轮将不胜感激。

Answer 1

我将此作为评论发布。

require 'prime'

a = (1..5).to_a
Hash[a.map{ |x| [x, x.prime?] }]
=> {1=>false, 2=>true, 3=>true, 4=>false, 5=>true}

Answer 2

以下内容将为您工作，使用Prime#prime?：

require 'prime'

a = [1,2,3,4,5]
Hash[a.zip(a.map(&Prime.method(:prime?)))]
# => {1=>false, 2=>true, 3=>true, 4=>false, 5=>true}

Answer 3

另一种解决方案：

a = [1,2,3,4,5]
results = {}

a.each {|i| results[i] = (1..i).map{|x| i/x.to_f % 1 == 0}.count(true) == 2}
puts results.inspect #=> {1 => false, 2 => true, 3 => true, 4 => false, 5 => true}