Zigzag数组 -- 面试宝典

最近在看面试宝典，其中看到一个题目说：输入一个正整数n，输出它的zigzag数组。

分析：书上给出了数学方面的思考然后给了代码。但是我感觉如果真是面试或者考试遇到的话，我这种笨脑袋肯定是想不出来的，因此最好的方式是观察数组，然后找到规律。观察可知，数字递增的方向只有4个：right， left-down， down， right-up。因此我们可以设置一个switch语句，然后每走一步，判断下步要走的方向。因为输出的是n*n的数组，因此矩阵的右下角的数字肯定不会超过n*n，用它作为结束条件即可。

(需要注意的是：刚开始做时忘记考虑矩阵赋值了一半时，右下角的怎么赋值。因此每个方向走完了，下一步都会有两种方向。不要忘记右下角的赋值。)

代码如下：

public class Test {
static int[][] zigzag(int n) {
int[][] a = new int[100][100];
int right = 0, left_down = 1, down = 2, right_up = 3;
int s = 0;
int it = n * n;
int flag = 0;
int i = 0, j = 0;
while(s <= it) {
a[i][j] = s;
switch (flag){
case 0: { //right
j++;
if(0 == i)
flag = 1;
else if(n-1 == i)
flag = 3;
break;
}
case 1: {//left-down
i++;
j--;
if(j == 0)
flag = 2;
else if(i == n-1)
flag = 0;
break;
}
case 2: {//down
i++;
if(0 == j)
flag = 3;
else if(n-1 == j)
flag = 1;
break;

}
case 3: {
i--;
j++;
if(0 == i)
flag = 0;
else if(n-1 == j)
flag = 2;
break;
}

}
s++;
}
return a;
}

public static void main(String[] args) {
// TODO Auto-generated method stub
int n = 7;
int[][]a =  zigzag(n);
for(int i=0; i<n; i++) {
for(int j=0; j<n; j++) {
System.out.print(a[i][j] + "  ");
}
System.out.println();
}

}

}