LeetCode 54 Spiral Matrix(螺旋矩阵)(Array)(*)

翻译

给定一个 mn 的矩阵(m行 n列),以螺旋状返回矩阵中的所有元素。

例如,给定以下矩阵

[
 [ 1, 2, 3 ],
 [ 4, 5, 6 ],
 [ 7, 8, 9 ]
]

你应该返回 [1,2,3,6,9,8,7,4,5] .

原文

Given a matrix of mn 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] .

分析

一开始我自以为很容易一眼就看出规律来,不过是用四个变量分别表示横纵坐标的起始点、再用两个变量表示当前的坐标,然后慢慢循环就Ok了。好吧,也许可以,但这种方法以后再也不想用了,实在是浪费时间又让人烦躁。

直接取看了别人的解法,其实有异曲同工,不过这个解法变量用的少了控制却更加巧妙。

向上的行是u,向下的行是d,向左的列是l,向右的列是r。

k是spiral的当前索引,不断往后走。

控制结束的是这4个方向的索引的大小比较。

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
        if (matrix.empty()) return {};
        int m = matrix.size(), n = matrix[0].size();
        vector<int> spiral(m * n);
        int u = 0, d = m - 1, l = 0, r = n - 1, k = 0;
        while (true) {
            // up
            for (int col = l; col <= r; col++) spiral[k++] = matrix[u][col];
            if (++u > d) break;
            // right
            for (int row = u; row <= d; row++) spiral[k++] = matrix[row][r];
            if (--r < l) break;
            // down
            for (int col = r; col >= l; col--) spiral[k++] = matrix[d][col];
            if (--d < u) break;
            // left
            for (int row = d; row >= u; row--) spiral[k++] = matrix[row][l];
            if (++l > r) break;
        }
        return spiral;
    }
};

LeetCode 59 Spiral Matrix II(螺旋矩阵II)(Array)

代码

Java

updated at 2016/09/04
    public List<Integer> spiralOrder(int[][] matrix) {
        List<Integer> spiral = new ArrayList<>();
        if (matrix.length < 1) return spiral;
        int m = matrix.length, n = matrix[0].length;        
        int u = 0, d = m - 1, l = 0, r = n - 1, k = 0;
        while (true) {
            // up
            for (int col = l; col <= r; col++) spiral.add(k++, matrix[u][col]);
            if (++u > d) break;
            // right
            for (int row = u; row <= d; row++) spiral.add(k++, matrix[row][r]);
            if (--r < l) break;
            // down
            for (int col = r; col >= l; col--) spiral.add(k++, matrix[d][col]);
            if (--d < u) break;
            // left
            for (int row = d; row >= u; row--) spiral.add(k++, matrix[row][l]);
            if (++l > r) break;
        }
        return spiral;
    }

你可能感兴趣的:(LeetCode,Algorithm,array,Matrix,54)