Lua 稀疏矩阵相乘

function mult(a, b)
	local c = {}
	for i = 1, #a do
		local resultline = {}					-- 即 c[i]
		for k, va in pairs(a[i]) do				-- 'va' 即 a[i][k]
			for j, vb in pairs(b[k]) do 		-- 'vb' 即 b[k][j]
				local res = (resultline[j] or 0) + va * vb
				resultline[j] = (res ~= 0) and res or nil
			end
		end
		c[i] = resultline
	end
	return c
end

  代码使用 pairs 来处理稀疏的元素,这样可以只访问非 nil 的元素。此外,代码还删去了结果中偶然为0的元素。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值