[第1000关] ZigZag遍历二维数组

ZigZag 是用在图像压缩中的一个算法

对于任意二维数组 ar[x_max][y_max]

按照下面的顺序遍历

算法思路：

1、对角线数=行数+列数-1

2、在对角线进行遍历

#include <log4ahk> ;https://www.autoahk.com/archives/36659
x_max := 5
y_max := 5
ar := create_matrix(5, 5)
log.info(ar)
;![](https://raw.githubusercontent.com/kazhafeizhale/pic/master/20220215204703.png)
;遍历数组
rtn_array := []

num_bias := x_max + y_max - 1
i := 1 ;当前行
j := 1 ;当前列
loop,% num_bias
{
    if(Mod(A_Index - 1, 2) == 0) ;偶数  左上遍历
    {
        while(i > 0 && j < y_max + 1)
        {
            rtn_array.Push(ar[i][j])
            i--
            j++
        }
        i++
        if(j > y_max)
        {
            i++
            j--
        }
    }
    else ;奇数
    {
        while(j > 0 && i < x_max + 1)
        {
            rtn_array.Push(ar[i][j])
            i++
            j--
        }
        j++
        if(i > x_max)
        {
            i--
            j++
        }
    }
}
log.info(rtn_array)

;输入行数和列数生成一个2维矩阵,并填充
;x 行数
;y 列数
create_matrix(x, y)
{
    array := []
    index := 1
    loop,% x
    {
        i := A_Index
        array.Push([])
        loop,% y
        {
            j :=  A_Index
            array[i][j] := index++
        }
    }
    return array
}