【剑指 offer】(Java) 顺时针打印矩阵

输入一个矩阵，按照从外向里以顺时针的顺序依次打印出每一个数字，例如，如果输入如下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.*;
public class Solution {
    public ArrayList<Integer> printMatrix(int [][] matrix) {
       ArrayList<Integer> arr = new ArrayList<>();
        int flag = 1; // 1-right  2-down 3-left 4-up
        int x = 0;
        int y = 0;
        // vis数组标记已经走过的点
        boolean[][] vis = new boolean[matrix.length][matrix[0].length];
        while (arr.size() < matrix.length * matrix[0].length) {
            // 判断是否越界  （已经走过的点再次遍历也视为越界
            if (x < 0 || x >= matrix.length || y < 0 || y >= matrix[0].length || vis[x][y]) {
                if (flag == 1) {
                    flag = 2; // down
                    y--; // 消除越界
                    x++; // 到达下一个位置的横坐标
                } else if (flag == 2) {
                    flag = 3; // left
                    x--;
                    y--;
                } else if (flag == 3) {
                    flag = 4; // up
                    y++;
                    x--;
                } else {
                    flag = 1; // right
                    x++;
                    y++; 
                }
            } else {
                arr.add(matrix[x][y]);
                vis[x][y] = true;
                // 更新下一个要访问的点
                if (flag == 1) {
                    y++;
                } else if (flag == 2) {
                    x++;
                } else if (flag == 3) {
                    y--;
                } else {
                    x--;
                }
            }
        }
        return arr;
    }
}