1

我正在尝试在 Zig 中分配 HashMap(u32, u1) 的二维数组：

fn alloc2d(comptime t: type, m: u32, n: u32, allocator: *Allocator) callconv(.Inline) ![][]t {
    const array = try allocator.alloc([]t, m);
    for (array) |_, index| {
        array[index] = try allocator.alloc(t, n);
    }
    return array;
}

fn free2d(comptime t: type, array: [][]t, allocator: *Allocator) callconv(.Inline) void {
    for (array) |_, index| {
        allocator.free(array[index]);
    }
    allocator.free(array);
}

test "Alloc 2D Array" {
    var gpa = std.heap.GeneralPurposeAllocator(.{}){};
    const allocator = &gpa.allocator;
    defer _ = gpa.deinit();

    const HashSet = std.AutoHashMap(u32, u1);
    var array = try alloc2d(*HashSet, 4, 4, allocator);
    defer free2d(*HashSet, array, allocator);

    for (array) |_, i| {
        for (array[i]) |_, j| {
            array[i][j] = &(HashSet.init(allocator));
        }
    }
    defer {
        for (array) |_, i| {
            for (array[i]) |_, j| {
                array[i][j].deinit();
            }
        }
    }
}

但是，当我测试它时，调试器会抛出一个段错误。

谁能告诉我发生了什么以及如何解决它？

非常感谢！

4

1 回答 1

2

我正在查看您的代码，乍一看它似乎符合您的预期；我不太确定你为什么将 a*HashSet而不仅仅是 a传递HashSet给你的函数：

...

var array = try alloc2d(HashSet, 4, 4, allocator);
defer free2d(HashSet, array, allocator);

for (array) |_, i| {
    for (array[i]) |_, j| {
        array[i][j] = HashSet.init(allocator);
    }
}

...

事实上，如果你这样做，一切都会如你所愿。

也就是说，我看不出您的版本不起作用的原因，所以我戳了一下，发现似乎正在发生的事情是您的每个项目array都使用相同的地址进行初始化，即&(HashSet.init(allocator))每次都返回相同的地址。我认为这就是为什么deinit调用出现段错误，内存被多次释放的原因。

如果您手动初始化数组中的每个元素，例如[0][0] = (HashSet.init(allocator)...etc一切似乎都有效。我不完全确定这里发生了什么，但可能是某种编译器优化在起作用，可能与泛型的工作方式有关。希望其他人会提供更好的答案。

有点不相关，但是一个简洁的特性Zig，你可以通过引用迭代切片，这有时更容易阅读：

for (array) |*outer| {
    for (outer.*) |*item| {
        item.* = <something>
    }
}
于 2021-03-15T18:09:49.427 回答