码迷,mamicode.com
首页 > 其他好文 > 详细

lintcode-medium-Spiral Matrix

时间:2016-04-06 11:25:15      阅读:177      评论:0      收藏:0      [点我收藏+]

标签:

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

 

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].

 

public class Solution {
    /**
     * @param matrix a matrix of m x n elements
     * @return an integer list
     */
    public List<Integer> spiralOrder(int[][] matrix) {
        // Write your code here
        
        List<Integer> res = new ArrayList<Integer>();
        
        if(matrix == null || matrix.length == 0 || matrix[0] == null || matrix[0].length == 0)
            return res;
        
        int m = matrix.length;
        int n = matrix[0].length;
        int layer = 0;
        
        while((layer < m - layer - 1) && (layer < n - 1 - layer)){
            
            for(int i = 0; i < n - layer - 1 - layer; i++)
                res.add(matrix[layer][layer + i]);
            
            for(int i = 0; i < m - layer - 1 - layer; i++)
                res.add(matrix[layer + i][n - layer - 1]);
            
            for(int i = 0; i < n - layer - 1 - layer; i++)
                res.add(matrix[m - layer - 1][n - layer - 1 - i]);
            
            for(int i = 0; i < m - layer - 1 - layer; i++)
                res.add(matrix[m - layer - 1 - i][layer]);
            
            layer++;
        }
        
        if(m > n){
            if(layer == n - 1 - layer){
                for(int i = layer; i <= m - layer - 1; i++)
                    res.add(matrix[i][layer]);
            }
        }
        else if(m < n){
            if(layer == m - 1 - layer){
                for(int i = layer; i <= n - layer - 1; i++)
                    res.add(matrix[layer][i]);
            }
        }
        else{
            if(layer == m - 1 - layer){
                res.add(matrix[layer][layer]);
            }
        }
        
        return res;
    }
    
}

 

lintcode-medium-Spiral Matrix

标签:

原文地址:http://www.cnblogs.com/goblinengineer/p/5358242.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!