本文介绍了如何在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语言免费学习笔记（深入）”；
-
导入必要的包：
-
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 语言内置的数据结构，使用起来更简洁高效。
示例：
如果输入以下内容：
1 2 3 4 5 6 7 8 9 10
程序将输出：
10 [1 2 3 4 5 6 7 8 9 10]
总结
通过使用 bufio 和 strconv 包，我们可以方便地在 Go 语言中实现类似 scanf 的输入解析功能。这种方式不仅灵活，而且可以根据实际需求进行定制，以支持不同的输入格式和数据类型。同时，需要注意错误处理和性能优化，以确保程序的稳定性和效率。
© 版权声明
文章版权归作者所有，未经允许请勿转载。
THE END
