LeetCode – Refresh – Spiral Matrix II

Do not forget to assign size value of matrix. This is very important!!!!

class Solution {
public:
    vector<vector<int> > generateMatrix(int n) {
        vector<vector<int> > result(n, vector<int>(n, 0));
        int count = 1;
        for (int i = 0; i < n/2; i++) {
            for (int j = i; j < n-i-1; j++) result[i][j] = count++;
            for (int j = i; j < n-i-1; j++) result[j][n-i-1] = count++;
            for (int j = n-i-1; j > i; j--) result[n-i-1][j] = count++;
            for (int j = n-i-1; j > i; j--) result[j][i] = count++;
        }
        if (n%2 == 1) result[n/2][n/2] = count;
        return result;
    }
};