牛客题练习

输入一个矩阵，按照从外向里以顺时针的顺序依次打印出每一个数字，例如，如果输入如下4 X 4矩阵： 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 则依次打印出数字1,2,3,4,8,12,16,15,14,13,9,5,6,7,11,10.

 

class Solution {
public:
    vector<int> printMatrix(vector<vector<int> > matrix) {
        vector<int> res;
        vector<vector<int>> dire={{0,1},{1,0},{0,-1},{-1,0}};
        int width=matrix[0].size(), height=matrix.size();
        int nums = width*height;
        vector<bool> tem(width, false);
        vector<vector<bool>> visited(height, tem);
        int round=0,i=0,j=0; //round用来控制dire中方向向量的选择
         while(0< nums){
            if(!visited[i][j]){
                        visited[i][j] = true;
                        cout<<matrix[i][j]<<",";
                        res.push_back(matrix[i][j]);
                        nums--;
                       if(nums==0) break;
                       if( j+dire[round][1]<0||i+dire[round][0]<0||i+dire[round[0]>= 
                           height  || j+dire[round][1]>=width||
                           visited[i+dire[round][0]][j+dire[round][1]])
                           {
                                round = (round+1)%4;  //改变方向
                           }
                        i += dire[round][0];
                        j += dire[round][1];
                      
            }
         } // while(0< nums)
         return res;
    }
};

大致思路：

设置方向数组0,1，-1，每次控制行列中一个的增减，按照右，下，左，上的方向循环直至所有节点都已访问（统计节点个数，num--），重点是方向的交替变化时间的判断：若出现越界（正向超，如>，负向超，如<）或者碰到已访问的元素，证明该换下一个方向了，round用来控制方向向量的选取，(round+1)%4的值更新用来表示循环到下一个方向，即：0,1,2,3,0,1,2,3...分别代表右，下，左，上，右，下，左，上...直至所有元素均已访问。结束、