C++實現(xiàn)LeetCode(59.螺旋矩陣之二)

更新時間：2021年07月16日 09:41:53 作者：Grandyang

這篇文章主要介紹了C++實現(xiàn)LeetCode(59.螺旋矩陣之二),本篇文章通過簡要的案例,講解了該項技術的了解與使用,以下就是詳細內(nèi)容,需要的朋友可以參考下

[LeetCode] 59. Spiral Matrix II 螺旋矩陣之二

Given a positive integer n, generate a square matrix filled with elements from 1 to n2 in spiral order.

Example:

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

此題跟之前那道 Spiral Matrix 本質上沒什么區(qū)別，就相當于個類似逆運算的過程，這道題是要按螺旋的順序來填數(shù)，由于給定矩形是個正方形，我們計算環(huán)數(shù)時用 n / 2 來計算，若n為奇數(shù)時，此時最中間的那個點沒有被算在環(huán)數(shù)里，所以最后需要單獨賦值，還是下標轉換問題是難點，參考之前 Spiral Matrix 的講解來轉換下標吧，參見代碼如下：

解法一：

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> res(n, vector<int>(n));
        int val = 1, p = n;
        for (int i = 0; i < n / 2; ++i, p -= 2) {
            for (int col = i; col < i + p; ++col)
                res[i][col] = val++;
            for (int row = i + 1; row < i + p; ++row)
                res[row][i + p - 1] = val++;
            for (int col = i + p - 2; col >= i; --col)
                res[i + p - 1][col] = val++;
            for (int row = i + p - 2; row > i; --row)    
                res[row][i] = val++;
        }
        if (n % 2 != 0) res[n / 2][n / 2] = val;
        return res;
    }
};

當然我們也可以使用下面這種簡化了坐標轉換的方法，博主個人還是比較推崇下面這種解法，不容易出錯，而且好理解，參見代碼如下：

解法二：

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> res(n, vector<int>(n));
        int up = 0, down = n - 1, left = 0, right = n - 1, val = 1;
        while (true) {
            for (int j = left; j <= right; ++j) res[up][j] = val++;
            if (++up > down) break;
            for (int i = up; i <= down; ++i) res[i][right] = val++;
            if (--right < left) break;
            for (int j = right; j >= left; --j) res[down][j] = val++;
            if (--down < up) break;
            for (int i = down; i >= up; --i) res[i][left] = val++;
            if (++left > right) break;
        }
        return res;
    }
};

到此這篇關于C++實現(xiàn)LeetCode(59.螺旋矩陣之二)的文章就介紹到這了,更多相關C++實現(xiàn)螺旋矩陣之二內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: