Zig编译报错：无法评估comptime表达式，求问题原因

我正在学习Zig语言，完成求第n个素数的练习。计划在编译期（comptime）生成所有素数并按需返回，当前代码如下：

const std = @import("std");
const mem = std.mem;

const buffer_size = 100000;

fn sieve(comptime buffer: []usize, comptime limit: usize) []usize {
    if (limit < 2)
        return buffer[0..0];
    const sqrt_limit = std.math.sqrt(limit);
    var bitset = std.StaticBitSet(limit + 1).initFull();
    var n: usize = 3;
    var i: usize = 1;
    
    
    buffer[0] = 2;
    while(n <= sqrt_limit): (n += 2) {
        if (!bitset.isSet(n))
            continue;

        buffer[i] = n;
        i += 1;
        var j = 2 * n;
        while (j <= limit): (j += i) {
            bitset.unset(j);
        }
    }
    while (n <= limit) : (n += 2) {
        if (!bitset.isSet(i))
            continue;

        buffer[i] = n;
        i += 1;
    }
    return buffer[0..i];
}


var primes_buffer: [buffer_size]usize = undefined;
const primes = sieve(&primes_buffer, buffer_size);

pub fn prime(allocator: mem.Allocator, number: usize) !usize {
    _ = allocator;
    return primes[number-1];
}

编译时出现如下错误：

nth_prime.zig:15:15: error: unable to evaluate comptime expression
nth_prime.zig:15:11: note: operation is runtime due to this operand
nth_prime.zig:39:21: note: called from here

请问为什么无法在编译期填充数组，且buffer[0] = 2;这一行存在什么问题？

核心问题原因 #

1. 编译期不能修改运行时变量：你用var声明的primes_buffer是全局运行时变量，但primes是const常量，会触发sieve函数在编译期执行。Zig明确禁止编译期代码修改运行时内存——编译期操作只能处理编译期可见的、不可变的内存块，或者专门的编译期分配内存。
2. buffer[0] = 2的本质问题：这行代码试图在编译期给运行时数组赋值，直接违反了上述规则，因此编译器抛出错误。
3. 额外逻辑错误：代码还有几处会导致编译/运行失败的问题：
   • std.math.sqrt(limit)返回f64类型，直接和usize的n比较会触发类型不匹配，编译期无法隐式转换浮点数到整数。
   • 第二个while循环中bitset.isSet(i)是错误的，应该判断bitset.isSet(n)，误用了索引变量代替当前数值。
   • 筛法内层循环j += i逻辑错误，埃氏筛的步长应该是n而非i，否则步长会随索引递增变化，无法正确标记非素数。

修正方案 #

要在编译期生成素数数组，需使用编译期分配的内存或comptime数组存储结果，以下是修正后的代码：

const std = @import("std");
const mem = std.mem;

const buffer_size = 100000;

// 编译期生成素数数组
fn sieve(comptime limit: usize) [_]usize {
    if (limit < 2) return [_]usize{};
    
    // 编译期转换平方根为usize类型
    const sqrt_limit = @as(usize, @intFromFloat(std.math.sqrt(@as(f64, limit))));
    var bitset = std.StaticBitSet(limit + 1).initFull();
    bitset.unset(0);
    bitset.unset(1);
    
    var n: usize = 2;
    while (n <= sqrt_limit) : (n += 1) {
        if (!bitset.isSet(n)) continue;
        var j = n * n;
        while (j <= limit) : (j += n) {
            bitset.unset(j);
        }
    }
    
    // 收集素数到编译期数组
    var primes_list: [buffer_size]usize = undefined;
    var count: usize = 0;
    n = 2;
    while (n <= limit) : (n += 1) {
        if (bitset.isSet(n)) {
            primes_list[count] = n;
            count += 1;
        }
    }
    
    return primes_list[0..count].*;
}

// 编译期生成素数常量数组
const primes = sieve(buffer_size);

pub fn prime(allocator: mem.Allocator, number: usize) !usize {
    _ = allocator;
    if (number < 1 or number > primes.len) {
        return error.InvalidInput;
    }
    return primes[number - 1];
}

关键修改点 #

• 将sieve改为直接返回编译期数组，不再依赖外部运行时缓冲区。
• 修正std.math.sqrt的类型转换，确保编译期可处理。
• 修复埃氏筛的步长、判断条件等逻辑错误。
• 添加输入合法性检查，避免数组越界。
• 所有操作在编译期完成，primes为编译期常量数组，运行时直接读取即可。

内容的提问来源于stack exchange，提问作者João Areias