多索引数组下连续0位对应矩阵预乘的伪代码实现问询

核心逻辑说明 #

使用两个索引完成遍历：一个主遍历索引i用于扫描整个标记数组，一个段起始索引start用于记录连续0标记段的起始位置，遇到非0标记直接存入结果数组，遇到连续0标记则合并对应区间的矩阵后再存入结果数组。

伪代码实现 #

输入：
1-based标记数组 flag_arr（元素为0或非0值）
1-based矩阵数组 M，长度与flag_arr一致
矩阵乘法运算方法 multiply(mat_a, mat_b) 返回两矩阵乘积

输出：
预处理后的矩阵数组 new_M

---
初始化 new_M = 空数组
n = length(flag_arr)
i = 1  // 主遍历索引

while i <= n:
    // 遇到非0标记，直接存入结果
    if flag_arr[i] != 0:
        append M[i] to new_M
        i = i + 1
        continue
    // 遇到0标记，找到连续0段的起止范围
    start = i  // 连续0段起始索引
    while i <= n and flag_arr[i] == 0:
        i = i + 1
    // 合并当前连续0段的所有矩阵
    temp_mat = M[start]
    for k from start+1 to i-1:
        temp_mat = multiply(temp_mat, M[k])
    // 合并结果存入新数组
    append temp_mat to new_M
end while

return new_M

效果验证 #

以你给出的示例标记数组[#,#,0,0,#,0,0,0,#]（#代表非0）为例，运行上述逻辑后得到的new_M结构如下：
[M[1], M[2], M[3]×M[4], M[5], M[6]×M[7]×M[8], M[9]]
完全符合预期需求，后续直接按顺序相乘new_M中所有元素即可得到最终总乘积。

内容的提问来源于stack exchange，提问作者stefano.sil