
本文介绍在算法竞赛场景下,如何优雅、健壮地从标准输入读取指定数量的整数(或 float/string),重点优化原始循环逻辑,加入错误处理,并提供可复用的通用模式。
在算法编程题中,常见输入格式为:首行给出数据个数 n,随后 n 行各含一个值(整数、浮点数或字符串)。原始代码虽能运行,但存在三个明显问题:
- 使用无限 for {} 循环 + 手动 break,语义不清晰;
- 忽略 fmt.Scanf 的返回错误,一旦输入异常(如非数字字符、EOF 提前)将导致静默失败或 panic;
- 变量作用域过大(如 input 在循环外声明),降低可读性与安全性。
以下是一个更简洁、健壮且符合 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]
}✅ 优势说明:
- 语义明确:使用 for i := 0; i
- 错误即刻检查:每次 Scanf 后立即校验 err,便于定位哪一行输入出错;
- 内存友好:make([]int, 0, count) 预分配底层数组容量,避免多次扩容;
- 作用域最小化:x 和 err 均限定在各自作用域内,符合 Go 最佳实践。
? 扩展支持 float / string:
只需替换类型与格式符即可:
- 浮点数:var x float64 + %f\n;
- 字符串(不含空格):var s string + %s\n;
- 含空格字符串:改用 bufio.Scanner(因 Scanf 对空白敏感):
import "bufio" scanner := bufio.NewScanner(os.Stdin) scanner.Scan() s := scanner.Text() // 安全读取整行
⚠️ 注意事项:
- %d\n 中的 \n 能跳过换行符,防止后续读取被残留换行干扰;若输入末尾无换行,可改用 %d 并手动调用 scanner.Scan() 处理;
- 竞赛环境通常保证输入合法,但加入错误处理是工程化思维的关键一步;
- 若需极致性能(如百万级输入),建议切换至 bufio.Reader + strconv 解析,避免 fmt.Scanf 的格式解析开销。
总结:用清晰的循环结构、及时的错误反馈和合理的内存预分配,让输入处理既可靠又易维护——这才是算法题中「正确」与「优雅」的统一。










