ruby 从 std 快速读取

Question

从 STDIN 读取 1000000 个字符（整数）并将其拆分为一个字符整数（而不是字符串）的数组的最快方法是什么？

123456 > [1,2,3,4,5,6]

Answer 1

到目前为止，我发现的最快方法如下：-

  gets.unpack("c*").map { |c| c-48}

以下是对大多数提供的解决方案进行基准测试的一些结果。这些测试使用 100,000 位数字的文件运行，但每个测试重复 10 次。

                                  user     system      total        real
each_char_full_array:         1.780000   0.010000   1.790000 (  1.788893)
each_char_empty_array:        1.560000   0.010000   1.570000 (  1.572162)
map_byte:                     0.760000   0.010000   0.770000 (  0.773848)
gets_scan                     2.220000   0.030000   2.250000 (  2.250076)
unpack:                       0.510000   0.020000   0.530000 (  0.529376)

这是生成它们的代码

#!/usr/bin/env ruby

require "benchmark"

MAX_ITERATIONS = 100000
FILE_NAME = "1_million_digits"

def build_test_file
  File.open(FILE_NAME, "w") do |f|
    MAX_ITERATIONS.times {|x| f.syswrite rand(10)}
  end
end

def each_char_empty_array
  STDIN.reopen(FILE_NAME)
  a = []
  STDIN.each_char do |c|
    a << c.to_i
  end
  a
end

def each_char_full_array
  STDIN.reopen(FILE_NAME)
  a = Array.new(MAX_ITERATIONS)
  idx = 0
  STDIN.each_char do |c|
    a[idx] = c.to_i
    idx += 1
  end
  a
end

def map_byte()
  STDIN.reopen(FILE_NAME)
  a = STDIN.bytes.map { |c| c-48 }
  a[-1] == -38 && a.pop
  a
end

def gets_scan
  STDIN.reopen(FILE_NAME)
  gets.scan(/\d/).map(&:to_i)
end


def unpack
  STDIN.reopen(FILE_NAME)
  gets.unpack("c*").map { |c| c-48}
end

reps = 10
build_test_file
Benchmark.bm(10) do |x|
  x.report("each_char_full_array: ") { reps.times {|y| each_char_full_array}}
  x.report("each_char_empty_array:") { reps.times {|y| each_char_empty_array}}
  x.report("map_byte:             ") { reps.times {|y| map_byte}}
  x.report("gets_scan             ") { reps.times {|y| gets_scan}}
  x.report("unpack:               ") { reps.times {|y| unpack}}
end

Answer 2

这应该相当快：

a = []
STDIN.each_char do |c|
  a << c.to_i
end

尽管一些粗略的基准测试显示这个黑客版本要快得多：

a = STDIN.bytes.map { |c| c-48 }

Answer 3

scan(/\d/).map(&:to_i)

这会将任何字符串拆分为整数数组，忽略任何非数字字符。如果您想从 STDIN 获取用户输入，请添加 gets：

gets.scan(/\d/).map(&:to_i)