码迷,mamicode.com
首页 > 其他好文 > 详细

LeetCode—*Spiral Matrix问题,主要是用到了方向矩阵,很创意

时间:2015-03-29 23:49:53      阅读:369      评论:0      收藏:0      [点我收藏+]

标签:spiral matrix   leetcoode   方向矩阵   

Given a matrix of m x n elements (m rows, n columns), return all elements of the matrix in spiral order.

For example,
Given the following matrix:

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

You should return [1,2,3,6,9,8,7,4,5].

这是一个螺旋排序的问题

这里遇到一个比较巧妙的方法,就是利用方向矩阵

class Solution {
public:
vector<int> spiralOrder(vector<vector<int> > &matrix) {
    vector<int> result;
    if(matrix.empty())
    {
        return result;
    }
    int m = matrix.size();
    int n = matrix[0].size();
    int direct = 0; //1:right,2:down,3:left,4:up
    int x[4] = {1,0,-1,0};
    int y[4] = {0,1,0,-1};
    
    int visitedRow = 0;
    int visitedCol = 0;
    int candidant = 0;
    int startx = 0;
    int starty = 0;
    int step = 0;
    while(true)
    {
        if(x[direct] == 0)//<表示做列的操作,关心的是行的数量
        {
            candidant = m-visitedRow;
        }
        else
        {
            candidant = n-visitedCol;//<做的是行的操作,关心的是列的数量
        }
        
        if(candidant <= 0) break;
        result.push_back(matrix[startx][starty]);
        step++;
        if(step == candidant) //<表示这个方向走完了
        {
            visitedRow += x[direct]==0? 0 : 1;
            visitedCol += y[direct]==0? 0 : 1;
            step = 0;
            direct++;
            direct = direct%4;
        }
        startx += y[direct]; //<x的位置和y方向的操作有关,注意这里的一个反向的操作
        starty += x[direct]; //<下一步的跳转过程
    }
    
    return result;
}
};


下面的方法就是比较简单的就是利用一个递归,每一次扫描一圈:

class Solution {
public:
vector<int> spiralOrder(vector<vector<int> > &matrix) {
    vector<int> result;
    if(matrix.empty())
    {
        return result;
    }
    int m = matrix.size();
    int n = matrix[0].size();
    
    //<通过观察,可以发现一定的循环规律
    SprialOrder(matrix,0,0,m,n,result);
    return result;
}
void SprialOrder(vector<vector<int> > &matrix,int startm,int startn,int m,int n,vector<int>& result)
{
    if(m <= 0 || n <= 0)
    {
        return;
    }
    for(int i = 0; i < n; i++)
    {
        result.push_back(matrix[startm][startn+i]);
    }
    for(int j = 1; j < m; j++)
    {
        result.push_back(matrix[startm+j][startn+n-1]);
    }
    for(int i = n-2; (i >= 0 && m >= 2);i--)
    {
        result.push_back(matrix[startm+m-1][startn+i]);
    }
    for(int j = m-3; j>=0; j--)
    {
        result.push_back(matrix[startm+j][startn]);
    }
    SprialOrder(matrix,startm+1,startn+1,m-2,n-2,result);
    return;
}
};


LeetCode—*Spiral Matrix问题,主要是用到了方向矩阵,很创意

标签:spiral matrix   leetcoode   方向矩阵   

原文地址:http://blog.csdn.net/xietingcandice/article/details/44731783

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!