假设有一个二维数组a[5][4]如下:
要求按照图中的箭头方向去遍历该二维数组,也就是ZigZag遍历二维数组,也称“之”字遍历二维数组。
算法设计如下:
(1)再计算该二维数组的对角线数目,如下图所示的绿色线条的数量,
通过观察二维数组可知,对角线数=行数+列数-1。
(2) 当对角线是偶数的时候,则从左下角向右上角方向遍历数组;当对角线是奇数的时候,则从右上角向左下角方向遍历数组。
(3)通过循环按照这个规则遍历数组,即可遍历二维数组的所有元素。
具体代码如下:
public static void zigzag(int[][] a){
int i=0,j=0;
int r = a.length; //行数
int c = a[0].length; //列数
int n = r+c-1; //对角线数
for (int k = 0; k < n; k++) {
if (k%2==0){ //偶数对角线
while (i >= 0 && j < c) {
System.out.print(a[i][j]+"\t");
i = i - 1;
j = j + 1;
}
i = i+1;
if (j>=c) {
j = c-1;
i = i+1;
}
}else { //奇数对角线
while (i < r && j >= 0) {
System.out.print(a[i][j]+"\t");
i = i + 1;
j = j - 1;
}
j = j+1;
if (i>=r) {
i = r-1;
j = j+1;
}
}
}
}