【算法题解：java】顺时针打印矩阵

题目

对于一个矩阵，请设计一个算法从左上角(mat[0][0])开始，顺时针打印矩阵元素。

给定int矩阵mat,以及它的维数nxm，请返回一个数组，数组中的元素为矩阵元素的顺时针输出。

测试样例：

[[1,2],[3,4]],2,2
返回：[1,2,4,3]

分析

注意只有一行或者一列时的特殊情形

代码

public class Printer {
    public int[] clockwisePrint(int[][] mat, int n, int m) {
        // write code here
        int [] arr = new int[n*m];
        int startX = 0;
        int endX = m - 1;
        int startY = 0;
        int endY = n - 1;
        int index = 0;

        while (startX <= endX && startY <= endY) {

            for (int i = startX; i <= endX; i++) {
                arr[index++] = mat[startY][i];
            }

            for (int i = startY + 1; i <= endY; i++) {
                arr[index++] = mat[i][endX];
            }

            if (endY > startY) {
                for (int i = endX - 1; i >= startX; i--) {
                    arr[index++] = mat[endY][i];
                }
            }

            if (endX > startX) {
                for (int i = endY - 1; i >= startY + 1; i--) {
                    arr[index++] = mat[i][startX];
                }
            }

            startX++;
            endX--;
            startY++;
            endY--;
        }
        return arr;
    }
}