如何创建任意大小的多维数组



我正在Zig中编写一个函数,该函数应该接受任意大小的多维数组。可能会有限制,但我无法提前对尺寸进行硬编码。

这里有一个例子:

const warn = @import("std").debug.warn;
fn printMap(map: []const [4]bool) void {
for (map) |row| {
for (row) |tile| {
warn("{}t", .{tile});
}
warn("n", .{});
}
}
pub fn main() !void {
const map = [_][4]bool{
[_]bool{ false, false, false, false },
[_]bool{ false, true, true, false },
[_]bool{ false, true, true, false },
[_]bool{ false, false, false, false },
};
printMap(map[0..]);
}

这会编译并运行,但如果我将函数签名更改为

fn printMap(map: []const []bool) void

我收到错误

expected type '[]const []bool', found '[]const [4]bool'

这可以用Zig表达吗?

您的map被声明为多维数组[4][4]bool,数组的长度是编译时类型的一部分(基于我对https://ziglang.org/documentation/master/#Slices)。

正如您从printMap中了解到的那样,如果您希望在运行时定义这些大小,则必须使用切片(带指针和长度的类型(,例如[][]bool

要使您的示例使用签名printMap(map: []const []bool),您可以执行以下操作:

var a = [_]bool{ false, false, true };
var b = [_]bool{ false, true, false };
const map = [_][]bool{
a[0..], // Make it a slice using slice syntax
&b,     // Or make it a slice by coercing *[N]T to []T
};
var c: []const []bool = map[0..]; // Pass it as a slice of slices
printMap(c);

要创建任意大小的多维切片数组,需要有一些缓冲区来存储数据。你可以使用一些静态内存,或者根据需要分配一些,一种方法可能是:

fn buildMap(x: u8, y: u8, allocator: *std.mem.Allocator) ![][]bool {
var map: [][]bool = undefined;
map = try allocator.alloc([]bool, x);
for (map) |*row| {
row.* = try allocator.alloc(bool, y);
}
return map;
}

应与printMap(map: []const []bool)配合使用。

另一种方法是使用一维数组/缓冲区,并对其进行适当的索引,但这并不能完全回答您的问题。我对这门语言还很陌生,所以可能有一些微妙之处我错过了。

最新更新