我怎样才能有效地从Go中的stdin读取数字(或者为什么fmt.Scanf效率太低)

时间:2015-05-01 08:07:09

标签: go

我开始在go中进行编程竞赛(只是为了学习语言)并且我惊讶地发现了

var T int
fmt.Scanf("%d", &T)

难以想象地慢。多慢?到read 10^5 integers它需要2.5秒(相比之下python在0.8秒内完成)。

那么为什么它如此缓慢?我应该如何正确阅读intuint64float64

1 个答案:

答案 0 :(得分:5)

如果你只有整数作为输入,这应该更快(不测试)

package main

import (
    "io/ioutil"
    "log"
    "os"
    "strconv"
)

func read() (int64, error) {
    b, err := ioutil.ReadAll(os.Stdin)
    if err != nil {
        return 0, err
    }

    // use strconv.ParseUint and strconv.ParseFloat in a similar way
    return strconv.ParseInt(string(b[:len(b)-1]), 10, 0)
}

func main() {
    i, err := read()
    if err != nil {
        log.Fatal(err)
    }

    println(i)
}

像这样运行

echo 123 | go run main.go

对于交互式输入,您可能想要使用bufio.NewReader,请参阅How to read input from console line?