Zig斐波那契生成器用户输入报错，求正确输入实现方案

问题描述 #

我编写了生成斐波那契数列的Zig代码，想要读取用户输入的数列项数，但编译时出现类型不匹配错误。希望能像C语言的scanf那样便捷获取输入，试过改用heap page_allocator但仍有错误，求正确的用户输入实现方法。

原代码如下：

const std = @import("std");
const stdout = std.io.getStdOut().writer();
const print = std.debug.print;

pub fn fibonacci(n: i32) i32 {
    if (n <= 1) {
        return n;
    }

    return fibonacci(n - 1) + fibonacci(n - 2);
}

pub fn main() !void {
    var num: i32 = 15;
    var i: i31 = 1;

    var buf: [10]i32 = undefined;

    const stdin = std.io.getStdIn().reader();

    try stdout.print("enter a number: ", .{});

    if (try stdin.readUntilDelimiterOrEof(buf[0..], '\n')) |user_input| {
        num = std.fmt.parseInt(i32, user_input, 10);
    } else {
        num = @as(i32, 0);
    }

    // creating a loop
    while (i <= num) {
        print("{} ", .{fibonacci(i)});

        i += 1;
    }
}

编译错误信息 #

> zig build-exe fibonacci.zig
fibonacci.zig:23:46: error: 期望类型 '[]u8'，找到 '*[10]i32'
    if (try stdin.readUntilDelimiterOrEof(buf[0..], '\n')) |user_input| {
                                          ~~~^~~~~
fibonacci.zig:23:46: note: 指针子类型 'i32' 无法转换为指针子类型 'u8'
fibonacci.zig:23:46: note: 无符号8位整数无法表示所有可能的有符号32位值
/home/alex/program/ziglang/lib/std/io.zig:187:18: note: 参数类型在此处声明
            buf: []u8,
                 ^~~~
referenced by:
    posixCallMainAndExit: /home/alex/program/ziglang/lib/std/start.zig:617:37
    _start: /home/alex/program/ziglang/lib/std/start.zig:424:40
    3 reference(s) hidden; use '-freference-trace=5' 查看所有引用

错误原因 #

1. 缓冲区类型不匹配：buf被定义为[10]i32，但readUntilDelimiterOrEof函数需要的是[]u8类型的字节缓冲区——用户输入的是文本字符串，每个字符对应1字节的u8值，而非i32数值。
2. 错误未处理：std.fmt.parseInt返回的是!i32错误联合类型，不能直接赋值给num，必须显式处理可能的解析错误。
3. 变量类型冗余：循环变量i使用i31没有必要，和num保持i32类型更统一。

修正后的代码 #

方案1：使用栈缓冲区（适合短输入） #

const std = @import("std");
const stdout = std.io.getStdOut().writer();
const print = std.debug.print;

pub fn fibonacci(n: i32) i32 {
    if (n <= 1) {
        return n;
    }
    return fibonacci(n - 1) + fibonacci(n - 2);
}

pub fn main() !void {
    var num: i32 = 15;
    var i: i32 = 1;

    // 改用u8类型的栈缓冲区存储输入文本
    var buf: [10]u8 = undefined;

    const stdin = std.io.getStdIn().reader();

    try stdout.print("enter a number: ", .{});

    if (try stdin.readUntilDelimiterOrEof(&buf, '\n')) |user_input| {
        // 用try处理parseInt的解析错误
        num = try std.fmt.parseInt(i32, user_input, 10);
    } else {
        num = 0;
    }

    while (i <= num) {
        print("{} ", .{fibonacci(i)});
        i += 1;
    }
    // 添加换行让输出更整洁
    try stdout.print("\n", .{});
}

方案2：使用分配器（适合不确定长度的输入） #

如果需要支持更长的输入，可以用Zig的分配器来动态分配缓冲区：

const std = @import("std");
const stdout = std.io.getStdOut().writer();
const print = std.debug.print;

pub fn fibonacci(n: i32) i32 {
    if (n <= 1) {
        return n;
    }
    return fibonacci(n - 1) + fibonacci(n - 2);
}

pub fn main() !void {
    var num: i32 = 15;
    var i: i32 = 1;

    // 初始化通用分配器，使用后要释放
    var gpa = std.heap.GeneralPurposeAllocator(.{}){};
    defer _ = gpa.deinit();
    const allocator = gpa.allocator();

    const stdin = std.io.getStdIn().reader();

    try stdout.print("enter a number: ", .{});

    // 用分配器自动处理缓冲区大小，最大限制1024字节
    if (try stdin.readUntilDelimiterOrEofAlloc(allocator, '\n', 1024)) |user_input| {
        defer allocator.free(user_input); // 确保内存被释放
        num = try std.fmt.parseInt(i32, user_input, 10);
    } else {
        num = 0;
    }

    while (i <= num) {
        print("{} ", .{fibonacci(i)});
        i += 1;
    }
    try stdout.print("\n", .{});
}

关键修正说明 #

• 缓冲区类型调整：将[10]i32改为[10]u8，匹配文本输入的字节类型要求。
• 错误处理：用try处理parseInt的解析错误，避免程序因无效输入崩溃。
• 分配器使用：动态分配版本需用defer释放内存，防止内存泄漏；readUntilDelimiterOrEofAlloc会自动分配足够的缓冲区存储输入。

内容的提问来源于stack exchange，提问作者Sourav Yadav