剑指offer刷题记录19——顺时针打印矩阵

题目描述

输入一个矩阵，按照从外向里以顺时针的顺序依次打印出每一个数字，例如，如果输入如下4 X 4矩阵： 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.

 

解法：

import java.util.ArrayList;
public class Solution {
    public ArrayList<Integer> printMatrix(int [][] matrix) {
        if(matrix == null) {
            return null;
        }
        int height = matrix.length;
        int width = matrix[0].length;
        int count = height * width;
        int direction = 0;
        int [][] flag = new int[height][width];
        ArrayList<Integer> result = new ArrayList<Integer>();
        //当前数字坐标
        int i = 0;
        int j = 0;
        while(count > 0) {
            result.add(matrix[i][j]);
            flag[i][j] = 1;
            if(direction % 4 == 0) {
                if(j == width - 1 || flag[i][j + 1] == 1) {
                    direction++;
                    i += 1;
                }
                else {
                    j += 1;
                }
            }            
            else if(direction % 4 == 1) {
                if(i == height - 1 || flag[i + 1][j] == 1) {
                    direction++;
                    j -= 1;
                }
                else {
                    i += 1;
                }
            }
            else if(direction % 4 == 2) {
                if(j == 0 || flag[i][j - 1] == 1) {
                    direction++;
                    i -= 1;
                }
                else {
                    j -= 1;
                }
            }
            else if(direction % 4 == 3) {
                if(i == 0 || flag[i - 1][j] == 1) {
                    direction++;
                    j += 1;
                }
                else {
                    i -= 1;
                }
            }            
            count--;
        }
        return result;
    }
}

写出来代码总觉得有点繁琐，而且用了另一个二维数组来表示是否遍历过该处数字，不过如果用圈数来判断的话又会多出一个变量，这样变量就太多了，看起来会很麻烦。

count表示矩阵中数字个数，当count减为0时表示数组已经遍历完成。

direction表示当前的遍历方向，0 1 2 3分别表示向右向下向左向上。

接下来就是方向变换条件的设置，注意将触及边界的条件放在短路运算符前面，以免产生数组越界的错误。