当输入来自重定向时，为什么 bufio.NewReader 不能正常工作？

Question

我目前正在尝试在网站 CodeChef 上解决 this challenge 手动输入输入时，我得到了问题中所述问题的预期答案。

但是当我尝试使用 cat in.txt | go run my_program.go 时，我的程序输出不同。

如您所见，我尝试使用 fmt.Printf 进行调试，看起来 bufio.NewReader 随机错过了来自管道的输入。

我想知道我做错了什么，以及是否有其他方法可以读取 Golang 中包含多个整数的整行。

谢谢你的帮助！

in.txt的内容：

手动输入时正常输出：

使用 cat in.txt | go run my_program.go

输入文本时输出调试消息

debug <1 2 3
>
1 1
debug <>
1 1
debug <>
1 1
debug <>
1 1

我的程序：


package main

import (
        "fmt"
        "os"
        "strings"
        "bufio"
        "strconv"
)

func main() {
    t, n := 0, 0
    fmt.Scanf("%d", &t)
    for i := 0; i < t; i++ {
            fmt.Scanf("%d\n", &n)
            v := make([]int, n)
            rd := bufio.NewReader(os.Stdin)
            text, _ := rd.ReadString('\n')
            fmt.Printf("debug <%s>\n", text)
            arr := strings.Split(strings.TrimSuffix(text, "\n"), " ")
            for k := 0; k < len(arr); k++ {
                    v[k], _ =  strconv.Atoi(arr[k])
            }
            fmt.Println(calc_intersect(v))
    }
}

func calc_intersect(v []int) (int, int) {
    smt, lgt := 100, 0
    scenario := make([]int, len(v))
    for sc := 0; sc < len(v); sc++  {
            infect := make([]bool, len(v))
            infect[sc] = true
            scenario[sc] = simulate(v, infect)
    }
    for i := 0; i < len(scenario); i++ {
            if scenario[i] <= smt {
                    smt = scenario[i]
            }
            if scenario[i] > lgt {
                    lgt = scenario[i]
            }
    }
    return smt, lgt
}

func simulate(v []int, infect []bool) int {
            var nb_inf int = 1
            pos := make([]int, len(v))
            for time := 0; time < 25; time++ {
                    for p := 0; p < len(v); p++ {
                            pos[p] = (v[p] * time) + p
                    }
                    for a := 0; a < len(v); a++ {
                            check_intersect(pos, infect, &nb_inf, a)
                    }
            }
            return nb_inf
}

func check_intersect(pos []int, infect []bool, nb_inf *int, a int) {
            for b := 0; b < len(pos); b++ {
                    if a == b {
                            continue
                    }
                    if pos[a] == pos[b] && (infect[a] || infect[b]) && !(infect[a] && infect[b]) {
                            *nb_inf += 1
                            infect[a], infect[b] = true, true
                    }
            }

}

Answer 1

您对 bufio.NewReader 的使用是错误的。每次通过 i 循环都会创建一个新的缓冲 reader。如果输入可用，reader 可能会提前读取，如果不可用，则 re-use 缓冲输入未使用的 reader。

只需将 rd := bufio.NewReader(os.Stdin) 移出循环即可解决问题。

当您通过标准输入从控制台读取时，您的代码可以正常工作，因为您键入的速度不够快，reader 无法尝试缓冲输入。

当输入来自重定向时，为什么 bufio.NewReader 不能正常工作？

why does bufio.NewReader does not work proprely when input come from redirection?

shell

input

pipe

go