leetCode 54.Spiral Matrix(螺旋矩陣) 解題思路和方法

leetCode 54.Spiral Matrix(螺旋矩陣) 解題思路和方法

Spiral Matrix

Given a matrix of m x n elements (m rows, n columns), return all elements of the matrix in spiral order.

For example,
Given the following matrix:

[
[ 1, 2, 3 ],
[ 4, 5, 6 ],
[ 7, 8, 9 ]
]
You should return [1,2,3,6,9,8,7,4,5].

思路：螺旋數組，需要控制輸出方向，我的實現是定義一個boolean數組，與數組大小相同，數據初始全為false.然後定義一個int變量表示方向，碰到數據邊界或者下一數據為true,則改變方向，直到越界或者全部為true。結束循環。

具體代碼如下：

public class Solution {
    public List spiralOrder(int[][] a) {
    	List list = new ArrayList();
    	if(a.length == 0 || a[0].length == 0){
    		return list;
    	}
        int i = 0;//行
        int j = 0;//列
        boolean[][] b = new boolean[a.length][a[0].length];

        int o = 0;//表示方向，0：右；1：下；2：左；3：上
        
        //在范圍內循環，超出范圍結束
        while(i < a.length && i >= 0 && j < a[0].length && j >= 0){
            if(b[i][j]){//如果已全部走完,結束循環
                break;
            }
            
            list.add(a[i][j]); //添加結果
            b[i][j] = true;//已添加的標記為true，表示已經添加
            
            switch(o){
                case 0://往右走的方向
                    if(j == a[0].length - 1 || b[i][j+1]){
                        o = 1;//走到最右或者已標記，方向往下走
                        i++;
                    }else{
                        j++;
                    }
                    break;
                    
                case 1:
                    if(i == a.length - 1 || b[i+1][j]){
                        o = 2;//走到最下或者已標記，方向往左走
                        j--;
                    }else{
                        i++;
                    }
                    break;
                case 2:
                    if(j == 0 || b[i][j-1]){
                        o = 3;//走到最左或者已標記，方向往上走
                        i--;
                    }else{
                        j--;
                    }
                    break;
                case 3:
                    if(i == 0 || b[i-1][j]){
                        o = 0;//走到最上或者已標記，方向往右走
                        j++;
                    }else{
                        i--;
                    }
                    break;
            }
        }
        return list;
    }
}