[LeetCode] 59. Spiral Matrix II

일명 달팽이문제! 1학년때 동아리에서 풀었던 기억이 새록새록 났다. 이 문제는 n의 수에 따라 nxn 모양의 matrix를 생성하고, 그 안에 달팽이 껍질 처럼 나선형으로 숫자를 채워줘야 한다.

 

여러 방법으로 풀 수 있으나, 여기서는 꺽여야 하는 모서리 점에 도달했는지를 조건으로 넣고, 도달했다면 방향을 꺾어주는 방식으로 풀이했다.

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> mat (n, vector<int>(n));
        int row = 0;
        int col = 0;
        int max = n-1;
        int direc = 0; // -> / v / <- / ^
        
        for (int i=0; i<n*n; i++){
            if (row == max && col == max || row == max && col == direc/4 || row == direc/4 && col == max || i!=0 && row == direc/4 + 1 && col == direc/4){
                direc++;
                if (direc%4 == 0) max--;
            }
            
            mat[row][col] = i + 1;
            if (direc%4 == 0) col++; //->
            else if (direc%4 == 1) row++; // v
            else if (direc%4 == 2) col--; // <-
            else if (direc%4 == 3) row--; // ^
        }
        return mat;
    }
};