算法刷题（17）_顺时针打印矩阵学习记录

算法刷题（17）_顺时针打印矩阵学习记录

【说明】

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

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

测试样例：

[[1,2],[3,4]],2,2

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

【算法实现】

package p4;

/**
 * 顺时针打印矩阵
 * @author Guozhu Zhu
 * @date 2018/7/29
 * @version 1.0
 *
 */
public class Test01 {
	
	public static void main(String[] args) {
		int[][] arr = {{1, 2, 3}, {4, 5, 6}, {7, 8, 9}, {10, 11, 12}};
		int[] res = Solution01(arr, 4, 3);
		for (int i : res) {
			System.out.println(i);
		}
	}
	
	public static int[] Solution01(int[][] arr, int n, int m) {
		int startX = 0;
        int endX = m-1;
        int startY = 0;
        int endY = n - 1;
        int index = 0;
        int[] result = new int[n * m];
        while (startX <= endX && startY <= endY) {
            //从左到右打印
            if(startX <= endX){
            for (int i = startX; i <= endX; i++) {
            	  result[index++] = arr[startY][i];
                }
            }
            // 从上往下打印
            if (startY < endY) {
                for (int i = startY + 1; i <= endY; i++) {
                    result[index++] = arr[i][endX];
                }
            }
            // 从右往左打印
            if (startX < endX && endY > startY) {
                for (int i = endX - 1; i >= startX; i--) {
                    result[index++] = arr[endY][i];
                }
            } 
            // 从下往上打印
            if (startY < endY && endX > startX) {
                for (int i = endY - 1; i >= startY + 1; i--) {
                    result[index++] = arr[i][startX];
                }
            }
            startX++;
            endX--;
            startY++;
            endY--;
        }
        return result;
	}

}