本文介绍在算法竞赛场景下,如何优雅、健壮地从标准输入读取指定数量的整数(或 float/string),重点优化原始循环逻辑,加入错误处理,并提供可复用的通用模式。
在算法编程题中,常见输入格式为:首行给出数据个数 n,随后 n 行各含一个值(整数、浮点数或字符串)。原始代码虽能运行,但存在三个明显问题:
以下是一个更简洁、健壮且符合 Go 习惯的写法:
package main
import "fmt"
func main() {
var count int
_, err := fmt.Scanf("%d\n", &count)
if err != nil {
panic(fmt.Errorf("failed to read count: %w", err))
}
nums := make([]int, 0, count) // 预分配容量,提升性能
for i := 0; i < count; i++ {
var x int
_, err := fmt.Scanf("%d\n", &x)
if err != nil {
panic(fmt.Errorf("failed to read element %d: %w", i+1, err))
}
nums = append(nums, x)
}
// 此处可继续处理 nums,例如输出验证:
fmt.Println(nums) // 示例输出:[78 42 99]
}✅ 优势说明:
? 扩展支持 float / string:
只需替换类型与格式符即可:
import "bufio" scanner := bufio.NewScanner(os.Stdin) scanner.Scan() s := scanner.Text() // 安全读取整行
⚠️ 注意事项:
fmt.Scanf 的格式解析开销。总结:用清晰的循环结构、及时的错误反馈和合理的内存预分配,让输入处理既可靠又易维护——这才是算法题中「正确」与「优雅」的统一。