Golang从终端读取多行不定长数据并赋值到数组

Golang 通过前序遍历和中序遍历构建树，需要从终端读取两行数据，并且事先不知道读取数据的长度，通过fmt.Scanf()实现

1.  
   func buildTree(preOrder []int, inorder []int) *TreeNode {
2.  
   if len(preOrder) == 0 {
3.  
   return nil
4.  
   }
5.  
   root := &TreeNode{Val: preOrder[0]}
6.  
   var i int
7.  
   for index, value := range inorder {
8.  
   if value == preOrder[0] {
9.  
   i = index
10.  
    break
11.  
    }
12.  
    }
13.  
    root.Left = buildTree(preOrder[1:i 1], inorder[:i])
14.  
    root.Right = buildTree(preOrder[i 1:], inorder[i 1:])
15.  
    return root
16.  
    }
17.  
     
18.  
    func main() {
19.  
    var num int
20.  
    var preOrder, inOrder []int
21.  
    for {
22.  
    _, err := fmt.Scanf("%d", &num)
23.  
    if err == nil {
24.  
    preOrder = append(preOrder, num)
25.  
    } else {
26.  
    break
27.  
    }
28.  
    }
29.  
    for {
30.  
    _, err := fmt.Scanf("%d", &num)
31.  
    if err == nil {
32.  
    inOrder = append(inOrder, num)
33.  
    } else {
34.  
    break
35.  
    }
36.  
    }
37.  
     
38.  
    fmt.Println(preOrder)
39.  
    fmt.Println(inOrder)
40.  
     
41.  
    root := buildTree(preOrder, inOrder)
42.  
    }

通过for循环与fmt.Scanf(),当Scanf读到换行符时，err不为nil，终端显示err为unexpected newline

通过err不为nil跳出for循环，第一行数据读完，由于只有两行数据，可以直接复制并修改第一个for循环，如果要读取多行数据的话，可以在嵌套一个for循环

另一种方法，使用bufio

1.  
   for {
2.  
   var num []int
3.  
   reader := bufio.NewReader(os.Stdin)
4.  
   strBytes, _, _ := reader.ReadLine()
5.  
   str := strings.Fields(string(strBytes))
6.  
   for i := range str {
7.  
   n, _ := strconv.Atoi(str[i])
8.  
   num = append(num, n)
9.  
   }
10.  
    fmt.Println(num)
11.  
    }

这篇好文章是转载于：学新通技术网