【剑指offer】顺时针打印矩阵

题目

输入一个矩阵，按照从外向里以顺时针的顺序依次打印出每一个数字，例如，如果输入如下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) {
        int i,j,k,dir,u;
        int m=matrix.size(),n=matrix[0].size();
        int le1=0,le2=n-1,h1=0,h2=m-1;
        vector<int> res;
        i=0;
        j=0;
        dir=0;
        for(k=0;k<m*n;k++){
            if(dir==0){//向右
                if(j>=le1 && j<=le2){
                    res.push_back(matrix[i][j]);
                    if(j==le2){
                        dir=(dir+1)%4;
                        i++;
                    }
                    else{
                        j++;
                    }
                }
            }
            else if(dir==1){//向下
                if(i>h1 && i<=h2){
                    res.push_back(matrix[i][j]);
                    if(i==h2){
                        dir=(dir+1)%4;
                        j--;
                    }
                    else
                        i++;
                }
            }
            else if(dir==2){//向左
                if(j>=le1 && j<le2){
                    res.push_back(matrix[i][j]);
                    if(j==le1){
                        dir=(dir+1)%4;
                        i--;
                    }
                    else{
                        j--;
                    }
                }
            }
            else{//向上

                if(i>h1 && i<h2){
                    res.push_back(matrix[i][j]);
                    if(i==h1+1){
                        dir=(dir+1)%4;
                        j++;
                        h1++;
                        h2--;
                        le1++;
                        le2--;
                    }
                    else
                        i--;
                }
            }
        }
        return res;
    }
};

精简的代码

class Solution {
public:
    vector<int> printMatrix(vector<vector<int> > matrix) {
        int i;
        int m=matrix.size(),n=matrix[0].size();
        int left=0,right=n-1,top=0,bot=m-1;
        vector<int> res;
        while(left<=right && top<=bot){
            for(i=left;i<=right;i++)
                res.push_back(matrix[top][i]);
            if(top<bot)
                for(i=top+1;i<=bot;i++)
                    res.push_back(matrix[i][right]);
            if(top<bot && left<right)
                for(i=right-1;i>=left;i--)
                    res.push_back(matrix[bot][i]);
            if(top+1<bot && left<right)
                for(i=bot-1;i>=top+1;i--)
                    res.push_back(matrix[i][left]);
            left++;
            right--;
            top++;
            bot--;
        }
        return res;
    }
};