剑指offer（21）：顺时针打印矩阵

题目描述

输入一个矩阵，按照从外向里以顺时针的顺序依次打印出每一个数字，例如，如果输入如下矩阵：
1 　　2 　3 　4
5 　　6 　7　 8
9 　10 　11 　12
13 　14 　15 　16
则依次打印出数字1,2,3,4,8,12,16,15,14,13,9,5,6,7,11,10.

分析

类似于蛇形矩阵（参见文章n阶蛇形矩阵)，在生成矩阵的时候直接进行打印输出，需要辅助flag数组，判断元素是否已经打印。从[0][0]位置开始依次向右、向下、向上、向左移动，每移动一次，进行打印，将该处的flag置为true。移动方向变化的条件是达到矩阵的边界或者当前元素的下一个元素已经打印输出过（为true）。

代码：

public class PrintMatrixClock {
    public static void main(String[] args) {
        int[][] arr = { { 1, 2, 3, 4 }, { 5, 6, 7, 8 }, { 9, 10, 11, 12 }, { 13, 14, 15, 16 } };
        printMatrix(arr);
    }

    public static void printMatrix(int[][] arr) {
        if (arr == null || arr.length <= 0)
            return;

        int rowCount = arr.length;
        int colCount = arr[0].length;

        boolean[][] flags = new boolean[rowCount][colCount];

        int x = 0, y = 0;
        int num = 1;

        // 首先打印输出第一个元素
        System.out.print(arr[0][0] + ", ");
        flags[0][0] = true;

        while (num < sum) {
            // 向右移动，
            // 直到到达边界，或者向右移动的下一个元素不为flag为true（已经打印过）
            while (y + 1 < colCount && flags[x][y + 1] == false) {
                num++;
                y++;
                System.out.print(arr[x][y] + ", ");
                flags[x][y] = true;
            }
            // 向下移动
            while (x + 1 < rowCount && flags[x + 1][y] == false) {
                num++;
                x++;
                System.out.print(arr[x][y] + ", ");
                flags[x][y] = true;
            }
            // 向左移动
            while (y - 1 >= 0 && flags[x][y - 1] == false) {
                num++;
                y--;
                System.out.print(arr[x][y] + ", ");
                flags[x][y] = true;
            }
            // 向上移动
            while (x - 1 >= 0 && flags[x - 1][y] == false) {
                num++;
                x--;
                System.out.print(arr[x][y] + ", ");
                flags[x][y] = true;
            }
        }
    }
}

牛客AC代码：

import java.util.ArrayList;
public class Solution {
    public ArrayList<Integer> printMatrix(int [][] matrix) {
       ArrayList<Integer> list = new ArrayList<Integer>();
       if(matrix == null || matrix.length <= 0)
            return list;

        int rowCount = matrix.length;
        int colCount = matrix[0].length;

        boolean[][] flags = new boolean[rowCount][colCount];


        int x = 0, y = 0;
        int num = 1;
        int sum = rowCount * colCount;
        list.add(matrix[0][0]);
        flags[0][0] = true;
        while(num < sum) {
            // 向右移动，
            // 直到到达边界，或者向右移动的下一个元素不为flag为true（已经打印过）
            while(y + 1 < colCount && flags[x][y + 1] == false) {
                num++;
                y++;
                list.add(matrix[x][y]);
                flags[x][y] = true;
            }
            // 向下移动
            while(x + 1 < rowCount && flags[x + 1][y] == false) {
                num++;
                x++;
                list.add(matrix[x][y]);
                flags[x][y] = true; 
            }
            // 向左移动
            while(y - 1 >= 0 && flags[x][y - 1] == false) {
                num++;
                y--;
                list.add(matrix[x][y]);
                flags[x][y] = true; 
            }
            // 向上移动
            while(x - 1 >= 0 && flags[x - 1][y] == false) {
                num++;
                x--;
                list.add(matrix[x][y]);
                flags[x][y] = true; 
            }   
        }
        return list;
    }
}

参考
1. 何海涛，剑指offer名企面试官精讲典型编程题（纪念版），电子工业出版社