探究多维数组与一维数组的转换是有实际意义的,
如:
1、给定一个立方体内坐标、一个顺序,问此坐标在立方体内按此顺序数,在第几个;
2、给定一堆物体,按某一顺序放在立方体的各坐标上,问最后一个会放在何处?
那么,三维要如何转为一维数组呢?
最容易想到的方式:维护一个递增变量,然后遍历,如下,
--定义一个三维数组和一个空的一维数组
local arr3 = {
{{0, 1, 2, 3}, {0, 1, 2, 3}, {0, 1, 2, 3}},
{{0, 1, 2, 3}, {0, 1, 2, 3}, {0, 1, 2, 3}},
};
local arr1 = {};
--各维长度,可由上面三维数组得到
local maxI, maxJ, maxK = 2, 3, 4;
--遍历转换
local n = 1;
function convert()
for i=1, maxI do
for j=1, maxJ do
for k=1, maxK do
arr1[n] = arr3[i][j][k];
n = n + 1;
end
end
end
end
这样确实没问题,但过于粗暴。
有没有更优雅的方式呢?答案是有!
上述方式中,一维数组的索引n, 是可以根据 maxI, maxJ, maxK 和 i, j, k 计算出来的。
对于三维数组,n 可表示为:
(i-1) * (maxJ * maxK) + (j-1) * maxK + k
这个式子是怎么得来的呢?
其实非常简单,只是因为Lua的数组索引是从1开始(三维数组的 i, j, k 和一维数组的n,均要由1开始), 绕了个弯子。。
如果是在其他语言中,数组索引从0开始,是什么样呢?
i * maxJ * maxK + j * maxK + k
上式可以看作是,从一点开始组合成直线,再组合成面,再组合成立体。。。
原来如此。。。更高纬度都可自行脑补了。
所以,在lua中,三维数组转一维数组的方法就变为了:
--定义一个三维数组和一个空的一维数组
local arr3 = {
{{0, 1, 2, 3}, {0, 1, 2, 3}, {0, 1, 2, 3}},
{{0, 1, 2, 3}, {0, 1, 2, 3}, {0, 1, 2, 3}},
};
local arr1 = {};
--各维长度,可由上面三维数组得到
local maxI, maxJ, maxK = 2, 3, 4;
--遍历
local function convert()
for i=1, maxI do
for j=1, maxJ do
for k=1, maxK do
local n = (i-1) * (maxJ * maxK) + (j-1) * maxK + k;
arr1[n] = arr3[i][j][k];
end
end
end
end
convert()
搞清上面的问题,想要一维转三维也是一样的。如下:
--定义一个一维数组和一个空的三维数组
local arr1 = {0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3};
local arr3 = {
{{}, {}, {}},
{{}, {}, {}},
};
--各维长度,可由上面三维数组得到
local maxI, maxJ, maxK = 2, 3, 4;
--遍历转换
local function convert()
for i=1, maxI do
for j=1, maxJ do
for k=1, maxK do
local n = (i-1) * (maxJ * maxK) + (j-1) * maxK + k;
arr3[i][j][k] = arr1[n];
end
end
end
end
convert(arr1)