矩阵”之“字打印

题目

对于一个矩阵，请设计一个算法，将元素按“之”字形打印。具体见样例。
给定一个整数矩阵mat，以及他的维数nxm，请返回一个数组，其中元素依次为打印的数字。

解法

区分依靠 i%2 == 0
具体代码实现如下：

import java.util.*;

public class Printer {
    public int[] printMatrix(int[][] mat, int n, int m) {
        // write code here
        int[] output =  new int[n*m];
        int w = 0;
        for(int i = 0;i < n;i++){
            for(int j = 0;j < m;j++){
                if(i%2 == 0){
                    output[w] = mat[i][j];
                }
                else{
                    output[w] = mat[i][m-1-j];
                }
                w++;
            }
        }
        return output;
    }
}