C++:Leetcode-螺旋矩阵-54题螺旋矩阵
此题的螺旋矩阵刚好和59题的螺旋矩阵思路相反。
此题需要多判断几种情况
重点还是把握循环不变量和模拟过程
题目
给你一个 m 行 n 列的矩阵 matrix ,请按照 顺时针螺旋顺序 ,返回矩阵中的所有元素。
思路分析
1.依次进行获取m*n矩阵元素,按上行,右列,下行,左列顺序
2.遵循循环不变量原则,左闭右开原则依次读取
3.行或列为1时,另外进行讨论
知识点
1、vector二维矩阵的定义及其赋值形式
2、vector二维矩阵的行数和列数的获取
代码
/*
Leetcode-螺旋矩阵-54题螺旋矩阵
54题和59题都是螺旋矩阵,但思路互为相反。
思路分析:
1.依次进行获取m*n矩阵元素,按上行,右列,下行,左列顺序
2.遵循循环不变量原则,左闭右开原则依次读取
3.行或列为1时,另外进行讨论
//明天接着写!!!!
//先撤退!!!
*/
#include "iostream"
#include "vector"
using namespace std;
class Solution
{
public:
vector<int> spiralOrder(vector<vector<int>> &matrix)
{
//存储元素
vector<int> res;
//变量
int startX = 0;
int startY = 0;
int offset = 1;
int i, j; //i行j列
int count = 1;
int loop;
//获取vector二维矩阵行数列数
int m = matrix.size(); //行数
int n = matrix[0].size(); //列数
//loop 圈数确定
//圈数为最小的行或列除以2
loop = (m >= n ? n : m) / 2;
//矩阵行为1时,另外进行判断
if (m == 1)
{
for (int k = 0; k < n; k++)
{
res.push_back(matrix[0][k]);
}
}
//矩阵列为1时
else if (n == 1)
{
for (int k = 0; k < m; k++)
{
res.push_back(matrix[k][0]);
}
}
//行列都不为1时
else
{
while (loop--)
{
//上行
for (j = startY; j < n - offset; j++)
{
res.push_back(matrix[startX][j]);
}
//右列
for (i = startX; i < m - offset; i++)
{
res.push_back(matrix[i][j]);
}
//下行
for (; j > startY; j--)
{
res.push_back(matrix[i][j]);
}
//左列
for (; i > startX; i--)
{
res.push_back(matrix[i][j]);
}
startX++;
startY++;
offset++;
}
//没有赋值完,即圈数循环后还有值没有进行赋值
if (res.size() != n * m)
{
//n维方阵,且为奇数
if (n == m && n % 2 == 1)
{
res.push_back(matrix[n / 2][n / 2]);
}
//n>m矩阵,列大于行
else if (n > m)
{
for (int k = startY; k <= n - offset; k++)
{
res.push_back(matrix[startX][k]);
}
}
//n<m矩阵,行大于列
else if (n < m)
{
for (int k = startX; k <= m - offset; k++)
{
res.push_back(matrix[k][startY]);
}
}
}
}
return res;
}
};
int main(int argc, char const *argv[])
{
int count = 1;
vector<vector<int>> matrix(5, vector<int>(3, 0)); //3行4列的矩阵,初始值全为0
//matrix 赋值初始化
for (int i = 0; i < 5; i++)
{
for (int j = 0; j < 3; j++)
{
matrix[i][j] = count++;
}
}
//获取vector二维矩阵行数列数
int m = matrix.size(); //行数
int n = matrix[0].size(); //列数
cout << "行数:" << m << " 列数:" << n << endl;
//matrix初始化输出测试
for (vector<vector<int>>::iterator it = matrix.begin(); it != matrix.end(); it++)
{
for (vector<int>::iterator itt = it->begin(); itt != it->end(); itt++)
{
cout << *itt << "\t";
}
cout << endl;
}
//测试!!
Solution s1;
vector<int> res = s1.spiralOrder(matrix);
for (vector<int>::iterator it = res.begin(); it != res.end(); it++)
{
cout << *it << endl;
}
return 0;
}
总结
熟悉vector二维的定义及其赋值,行数列数的获取。
模拟过程的熟悉,循环不变量的熟悉。