java 顺时针打印矩阵

题目描述

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

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

测试样例：

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

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

代码如下：

import java.util.*;

public class Printer {
    public int[] clockwisePrint(int[][] mat, int n, int m) {
        int x1=0,y1=0;
        int x2=n-1,y2=m-1;
        int i,j;
        int len=n*m;
        int arr[]=new int[len];
        int flag=1;
        int count=0;
        while(true){
            if(flag==1){
                for(j=y1;j<=y2;j++){
                    arr[count++]=mat[x1][j];
                }
                if(count==len)break;
                flag=2;
                x1++;
            }else if(flag==2){
                for(i=x1;i<=x2;i++){
                    arr[count++]=mat[i][y2];
                }
                if(count==len)break;
                flag=3;
                y2--;
            }else if(flag==3){
                for(j=y2;j>=y1;j--){
                    arr[count++]=mat[x2][j];
                }
                if(count==len)break;
                flag=4;
                x2--;
            }else{
                for(i=x2;i>=x1;i--){
                    arr[count++]=mat[i][y1];
                }
                if(count==len)break;
                flag=1;
                y1++;
            }
        }
        return arr;
    }
}