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

Question

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

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

正如您所看到的，我尝试使用调试fmt.Printf，看起来像是bufio.NewReader随机错过了管道的输入。

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

谢谢你帮助我！

内容in.txt：

4
3
1 2 3
3
3 2 1
3
0 0 0
3
1 3 2

手动输入输入时的正常输出：

1 1
3 3
1 1
1 2

使用输入文本时输出带有调试消息 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循环都会创建一个新的缓冲读取器。如果输入可用，阅读器可能会提前阅读，并且当您不重复使用阅读器时，缓冲输入未使用。

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

当您通过 stdin 从控制台读取时，您的代码会起作用，因为您的输入速度不够快，无法让读者尝试缓冲输入。