Go语言中实现类似scanf()功能的输入处理-Golang-PHP中文网

Go语言中实现类似scanf()功能的输入处理

聖光之護

发布： 2025-07-13 16:32:14

原创

792人浏览过

go语言中实现类似scanf()功能的输入处理

本文旨在介绍如何在Go语言中实现类似于C语言中scanf()函数的输入处理功能。通过使用bufio包、os包、strings包和strconv包，我们可以从标准输入或其他io.Reader中读取数据，并将其转换为所需的数据类型，例如整数。本文将提供一个可复用的scanf()函数，并详细解释其实现原理和使用方法，帮助读者更好地理解和掌握Go语言的输入处理技巧。

在Go语言中，没有直接等价于C语言scanf()的函数。但是，我们可以利用Go的标准库来实现类似的功能。以下是一个示例，展示了如何从标准输入读取整数，并提供详细的解释。

使用 bufio 和 strconv 实现 scanf() 功能

该方法的核心思想是使用 bufio.NewReader 从 io.Reader (例如 os.Stdin) 中读取数据，然后使用 strings.Fields 将读取的字符串分割成字段，最后使用 strconv.Atoi 将字段转换为整数。

package main

import (
    "bufio"
    "fmt"
    "io"
    "os"
    "strconv"
    "strings"
    "container/vector"
)

func main() {
    n := scanf(os.Stdin)
    fmt.Println()
    fmt.Println(len(n), n)
}

func scanf(in io.Reader) []int {
    var nums vector.IntVector
    rd := bufio.NewReader(in)
    str, err := rd.ReadString('\n')
    for err != io.EOF {
        fields := strings.Fields(str)
        for _, f := range fields {
            if i, err := strconv.Atoi(f); err == nil {
                nums.Push(i)
            }
        }
        str, err = rd.ReadString('\n')
    }
    return nums
}

登录后复制

代码解释：

立即学习“go语言免费学习笔记（深入）”；

package main 和 import 语句: 声明包为 main，并导入必要的包，包括 bufio（用于缓冲读取），fmt（用于格式化输出），io（用于输入输出接口），os（用于操作系统接口），strconv（用于字符串转换），strings（用于字符串操作），container/vector（用于动态数组）。
main 函数:
- 调用 scanf 函数，从标准输入读取整数。
- 打印读取的整数的数量和内容。
scanf 函数:
- func scanf(in io.Reader) []int: 定义一个名为 scanf 的函数，它接受一个 io.Reader 类型的参数 in，并返回一个整数切片。
- var nums vector.IntVector: 声明一个名为 nums 的整数切片，用于存储读取的整数。
- rd := bufio.NewReader(in): 创建一个 bufio.Reader，用于缓冲读取输入。
- str, err := rd.ReadString('\n'): 从输入中读取一行字符串，直到遇到换行符 \n。
- for err != io.EOF: 循环读取输入，直到遇到文件结束符 io.EOF。
- fields := strings.Fields(str): 使用 strings.Fields 函数将读取的字符串分割成字段，默认以空格分割。
- for _, f := range fields: 遍历分割后的字段。
- if i, err := strconv.Atoi(f); err == nil: 尝试将每个字段转换为整数。如果转换成功（err == nil），则将整数添加到 nums 切片中。
- str, err = rd.ReadString('\n'): 读取下一行字符串。
- return nums: 返回包含所有读取的整数的切片。