Go语言中实现类似scanf功能的函数-Golang-PHP中文网

Go语言中实现类似scanf功能的函数

霞舞

发布： 2025-07-13 16:42:24

原创

990人浏览过

go语言中实现类似scanf功能的函数

本文介绍了如何在Go语言中实现类似C语言中scanf函数的输入处理功能。通过bufio包和strconv包，我们可以从标准输入或其他io.Reader中读取数据，并将字符串转换为整数或其他类型，从而实现灵活的输入解析。文章提供了一个具体的示例，并对代码进行了优化，使其更加简洁高效。

在Go语言中，并没有直接等价于C语言scanf的函数。但是，我们可以利用Go的标准库来实现类似的功能，从标准输入或其他输入源读取数据并进行解析。以下提供一种实现方式，并进行详细解释。

使用 bufio 和 strconv 实现 scanf 功能

核心思路是使用 bufio 包读取输入，然后使用 strconv 包将字符串转换为需要的类型。

package main

import (
    "bufio"
    "fmt"
    "io"
    "os"
    "strconv"
    "strings"
)

func scanf(in io.Reader) []int {
    var nums []int // 使用切片代替 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 = append(nums, i)
            }
        }
        str, err = rd.ReadString('\n')
    }
    return nums
}

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

登录后复制

代码解释：

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

导入必要的包：
- bufio：用于缓冲读取，提高效率。
- fmt：用于输出。
- io：用于输入输出接口。
- os：用于访问操作系统功能，例如标准输入。
- strconv：用于字符串转换。
- strings：用于字符串处理。
scanf 函数：
- 接收一个 io.Reader 接口作为输入源，这意味着可以从任何实现了 io.Reader 接口的对象读取数据，例如标准输入 os.Stdin、文件等。
- 使用 bufio.NewReader 创建一个缓冲读取器，提高读取效率。
- 使用 rd.ReadString('\n') 读取一行数据，直到遇到换行符。
- 循环读取，直到遇到 io.EOF（文件结束符）错误。
- 使用 strings.Fields(str) 将读取的行按空格分割成字符串切片。
- 遍历字符串切片，尝试使用 strconv.Atoi(f) 将每个字符串转换为整数。
- 如果转换成功（err == nil），则将整数添加到 nums 切片中。
- 最后返回包含所有成功转换的整数的切片。
main 函数：
- 调用 scanf 函数，并将标准输入 os.Stdin 作为输入源。
- 打印读取到的整数切片的长度和内容。

注意事项：

错误处理： 代码中对 strconv.Atoi 的错误进行了简单的忽略，实际应用中应该进行更完善的错误处理，例如记录错误日志或返回错误信息。
输入格式： 此代码假设输入是以空格分隔的整数，如果输入格式不符合要求，可能会导致转换失败。可以根据实际需求修改代码以支持其他输入格式。
类型转换： 可以根据需要使用 strconv 包中的其他函数将字符串转换为其他类型，例如 strconv.ParseFloat、strconv.ParseBool 等。
切片的使用： 代码中使用切片 []int 代替了原代码中的 container/vector 包，因为切片是 Go 语言内置的数据结构，使用起来更简洁高效。

示例：

如果输入以下内容：