Go 语言中使用 bufio.NewReader 等函数实现快读

Go 语言中如果标准输入的数据规模在 10 ^ 5 或者更大，那么如果直接使用 fmt.Scan() 和 fmt.Println() 函数读取数据和输出数据会非常慢，如果算法的时间复杂度更大一点就会超时，这个时候就需要借助于 bufio 包中的相关方法优化读取数据和输出数据的速度，可以使用 bufio.NewReader()，bufio.NewWriter()，fmt.Fscan()，fmt.Fprintln() 函数优化输入输出数据的速度，我们可以使用 Go 语言生成一个 10 ^ 5 规模的数据简单测试一下（因为我们自己输入和系统输入的速度还是有差别的，自己手动复制粘贴的速度肯定比系统输入要慢得多，这里只是简单测试一下）：可以发现使用 fmt.Scan()，fmt.Println() 函数大概需要 11s 左右，而使用bufio.NewReader()，bufio.NewWriter()，fmt.Fscan()，fmt.Fprintln() 函数大概需要 9s 左右，如果数据量更大的时候那么最终两者的处理时间是相差 5s~6s 左右，所以差别就很大了：

1. 生成 10 ^ 5 规模的 0~9 之间的数字：

package main

import (
	"os"
	"strconv"
)

func main() {
    // create函数创建一个不存在的文件
	file, _ := os.Create("data.txt")
	// 函数返回的时候关闭文件句柄
	defer file.Close()
	for i := 0; i < 100000; i++ {
		file.WriteString(strconv.Itoa(i%10) + "\n")
	}
}

2. 使用 fmt.Scan() 和 fmt.Println() 函数：

package main

import (
	"fmt"
	"time"
)

func main() {
	start := time.Now()
	const N = 100000
	var a [N]int
	for i := 0; i < N; i++ {
		fmt.Scan(&a[i])
	}
	end := time.Now()
	fmt.Println(end.Sub(start))
}

3. 使用 bufio.NewReader()，bufio.NewWriter()，fmt.Fscan()，fmt.Fprintln() 函数：

package main

import (
	"bufio"
	"fmt"
	"io"
	"os"
	"time"
)

func run(r io.Reader, w io.Writer) {
	start := time.Now()
	in := bufio.NewReader(r)
	out := bufio.NewWriter(w)
	defer out.Flush()
	const N = 100000
	var a [N]int
	for i := 0; i < N; i++ {
        // fmt.Fscan()函数从标准输入in中读取数据
		fmt.Fscan(in, &a[i])
	}
	end := time.Now()
    // fmt.Fprintln() 函数从标准输出out中写入数据
	fmt.Fprintln(out, end.Sub(start))
}

func main() {
    // 标准输入和标准输出
	run(os.Stdin, os.Stdout)
}