当输入来自重定向时，为什么 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

}

Answer 1

你的使用bufio.NewReader是错误的。每次通过i循环创建一个新的缓冲阅读器。如果输入可用，则阅读器可能会提前读取，并且当您不重新使用未使用缓冲输入的阅读器时。只需移出rd := bufio.NewReader(os.Stdin)循环即可解决问题。当您通过标准输入从控制台读取时，您的代码可以工作，因为您的输入速度不够快，读者无法尝试缓冲输入。