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

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

func buildTree(preOrder []int, inorder []int) *TreeNode {
	if len(preOrder) == 0 {
		return nil
	}
	root := &TreeNode{Val: preOrder[0]}
	var i int
	for index, value := range inorder {
		if value == preOrder[0] {
			i = index
			break
		}
	}
	root.Left = buildTree(preOrder[1:i+1], inorder[:i])
	root.Right = buildTree(preOrder[i+1:], inorder[i+1:])
	return root
}

func main() {
	var num int
	var preOrder, inOrder []int
	for {
		_, err := fmt.Scanf("%d", &num)
		if err == nil {
			preOrder = append(preOrder, num)
		} else {
			break
		}
	}
	for {
		_, err := fmt.Scanf("%d", &num)
		if err == nil {
			inOrder = append(inOrder, num)
		} else {
			break
		}
	}

	fmt.Println(preOrder)
	fmt.Println(inOrder)

	root := buildTree(preOrder, inOrder)
}

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

通过err不为nil跳出for循环,第一行数据读完,由于只有两行数据,可以直接

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值