螺旋矩阵
螺旋矩阵是什么问题?
螺旋矩阵是将一个二维数组按照顺时针转着圈着写进一个以为数组,或者是反过来。
螺旋矩阵问题解决的思路是什么?
按照左闭右开不变量原则 转n/2圈,转一圈里面有四个循环,将二维数组的值放入一维数组中或者相反。
变量 计数变量用来记录往二维数组中放置的数,istart和jstart用来放每次转圈的起始位置,每次循环之后两个都+1;一个offset来控制loop的大小,每次循环之后loop+1,如果n为奇数就需要判断一下单独处理最后一个值。
代码
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
vector<vector<int>> array(n, vector<int>(n));
int x = 0, y = 0;
int count = 1;
int offset = 1;
int i, j;
int loop = n / 2;
int middle = n / 2;
while(loop--) {
i = x;
j = y;
for (j; j < n - offset; j++) {
array[i][j] = count++;
}
for (i; i < n - offset; i++) {
array[i][j] = count++;
}
for (j; j >= offset; j--) {
array[i][j] = count++;
}
for (i; i >= offset; i--) {
array[i][j] = count++;
}
y++;
x++;
offset++;
}
if (n % 2 == 1) {
array[middle][middle] = n * n;
}
return array;
}
};
变形
m*n的二维矩阵,就要求loop变了,min(m,n)/2,while(loop--)中的赋值操作也变了,最后额外处理也变了,需要讨论m为奇数和n为奇数的情况,还需要一个for循环来赋值,别忘了根据题目给出的条件判断特殊情况。
代码
class Solution {
public:
vector<int> spiralOrder(vector<vector<int>>& matrix) {
int m = matrix.size();
int n = matrix[0].size();
int offset = 1;
int loop = min(m, n)/2;
int iStart = 0;
int jStart = 0;
int i, j;
vector<int> result;
if (m == 1) {
return matrix[0];
}
if (n == 1) {
for (int i = 0; i < m; i++) {
result.push_back(matrix[i][0]);
}
return result;
}
while (loop--) {
for ( j = jStart; j < n - offset; j++) {
result.push_back(matrix[iStart][j]);
}
for ( i = iStart; i < m - offset; i++) {
result.push_back(matrix[i][j]);
}
for (; j > jStart; j--) {
result.push_back(matrix[i][j]);
}
for (; i > iStart; i--) {
result.push_back(matrix[i][j]);
}
offset++;
iStart++;
jStart++;
}
if (m%2) {
for (j = jStart; j < n - offset + 1; j++ ) {
result.push_back(matrix[iStart][j]);
}
}
else {
if(n%2) {
for (i = iStart; i < m - offset + 1; i++) {
result.push_back(matrix[i][jStart]);
}
}
}
return result;
}
};
语法
要记住的是二维vector怎么创建,二维vector的长宽怎么获得,怎么return 空集合,还要学会push_back