【LeetCode】Spiral Matrix解题笔记

1、Spiral Matrix

题意：提供一个m*n的矩阵（m行，n列），上右下左的次数，一圈一圈的次序输出该矩阵。例如

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

输出结果为[1,2,3,6,9,8,7,4,5].

分析：一圈一圈的输出，其实就是将矩阵四周一层一层的剥下来，总共有多少层，就循环多少次，[1]这种只有一个数字的，算一层。

这里，m==n的时候，其实就是方阵，跟问题2的算法是一样的。m!=n的时候，这个算起来就有些复杂了，尤其是碰到m*1或1*n的时候，就需要格外小心边界条件的判定，这里很容易引起越界。

代码：

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int> > &matrix) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
		int m = matrix.size();
		if (m == 0)
		{
			return vector<int>();
		}
		
		int n = matrix[0].size();
		vector<int> vec;
		int levelm = (m+1)/2;
		int leveln = (n+1)/2;
		for (int a = 0, b = 0; a < levelm&&b < leveln; ++a, ++b)
		{
			for (int c = 0; c < 4; ++c)
			{
				int j;
				switch(c)
				{
				case 0:
					for (j = a; j <n - a; ++j)
					{
						vec.push_back(matrix[a][j]);
					}
					break;
				case 1:
					for (j = b+1; j < m-b && n-a >0; ++j)
					{
						vec.push_back(matrix[j][n-a-1]);
					}
					break;
				case 2:
					for (j = n-a-2; j >= a && m-b >0 && m-b-1 != a; j--)
					{
					    
						vec.push_back(matrix[m-b-1][j]);
					}
					break;
				case 3:
					for (j = m-b-2; j >= b+1 && b != n-a-1; j --)
					{
						vec.push_back(matrix[j][b]);
					}
					break;
				default:
					break;
				}
			}
		}

		return vec; 
    }
};

代码运行时间为8ms。

这里需要注意的就是边界条件：

b != n-a-1

m-b-1 != a

a的判断用来防止只有一行的时候，这一行的重复遍历。b的判断用来防止只有1列的时候，列的重复遍历。

2、Spiral Matrix II 

题意：给出一个数字n，输出一个n*n的方阵，方阵的排布同问题1中的排布方式，上右下左顺序输出。

分析：这个是一个经典的题目，有了题目1作为基础，其实这个比1还要简单的，因为是方阵，所以矩阵一定会有四条边，也就不需要区分遍历的时候的行列index。

代码：

class Solution {
public:
    vector<vector<int> > generateMatrix(int n) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        	int level = (n+1)/2;
		vector<vector<int>> ret;
		ret.resize(n);
		for (int i = 0; i < n; ++i)
		{
			ret[i].resize(n);
		}

		int x = 1;
		for (int a = 0; a < level; ++a)
		{
			for (int b = 0; b < 4; ++b)
			{
				switch(b)
				{
				case 0:
					for (int j = a; j < n - a; ++j)
					{
						ret[a][j] = x ++;
					}
					break;
				case 1:
					for (int j = a+1; j < n-a; ++j)
					{
						ret[j][n-a-1] = x ++;
					}
					break;
				case 2:
					for (int j = n-a-2; j >= a; j--)
					{
						ret[n-a-1][j] = x ++;
					}
					break;
				case 3:
					for (int j = n-a-2; j >= a+1; j --)
					{
						ret[j][a] = x ++;
					}
					break;
				default:
					break;
				}
			}
		}

		return ret;
        
    }
};

这个代码AC的时间是32ms，时间有些长，笔者正在研究是哪里耗时了。