“之”字形打印矩阵

题目描述

给定一个矩阵matrix，按照“之”字形的方式打印这 个矩阵，例如:
1 2 3 4
5 6 7 8
9 10 11 12
“之”字形打印的结果为:1，2，5，9，6，3，4，7，10，11， 8，12
要求： 额外空间复杂度为O(1)。

思路

这题也应使用宏观结构的思想，不要局限与下标要怎么变。
首先给两个点A、B,初始位置都在0,0点。A向右移动，B向左移动，每一个状态都可以以A、B为两端点画一条对角线，依次从上往下或从下往上打印对角线。

如果B已经到最后一行，则向右走。

如果A到最右边了，则向下走。

代码

public class ZigZagPrintMatrix {

    public static void printMatrixZigZag(int[][] matrix){
        int aR = 0;
        int aC = 0;
        int bR = 0;
        int bC = 0;
        int endR = matrix.length - 1;
        int endC = matrix[0].length - 1;
        boolean fromUp = false;
        while ( aR != endR + 1 ){
            print(matrix,aR,aC,bR,bC,fromUp);
            aR = aC == endC ? aR + 1 : aR;
            aC = aC == endC ? aC : aC + 1;
            bC = bR == endR ? bC + 1 : bC;
            bR = bR == endR ? bR : bR + 1;
            fromUp = !fromUp;
        }
    }

    public static void print(int[][] matrix,int aR,int aC,int bR,int bC,boolean fromUp){
        if (fromUp){
            while (aR != bR + 1){
                System.out.print(matrix[aR++][aC--]+" ");
            }
        }else {
            while (bR != aR - 1){
                System.out.print(matrix[bR--][bC++]+" ");
            }
        }

    }

    public static void main(String[] args) {
        int[][] matrix = { { 1, 2, 3, 4 }, { 5, 6, 7, 8 }, { 9, 10, 11, 12 } };
        printMatrixZigZag(matrix);

    }
}