螺旋矩阵的上下左右四指针解法

最新推荐文章于 2023-03-18 23:33:10 发布

Mr.RottenPeach

最新推荐文章于 2023-03-18 23:33:10 发布

阅读量250

点赞数

分类专栏： Coding 文章标签： python 开发语言后端

本文链接：https://blog.csdn.net/racesu/article/details/109682636

版权

Coding 专栏收录该内容

9 篇文章 0 订阅

订阅专栏

《螺旋矩阵的上下左右四指针解法》

本文记录一下LeetCode上的两道螺旋矩阵题目的解法，感觉比较有趣，记录一下。

螺旋矩阵
Runtime:0 ms, faster than 100.00% of C++ online submissions.
Memory Usage:7 MB, less than 24.87% of C++ online submissions.

螺旋矩阵II
Runtime:0 ms, faster than 100.00% of C++ online submissions.
Memory Usage:6.8 MB, less than 21.52% of C++ online submissions.

Key Words：螺旋矩阵、螺旋矩阵II、上下左右、C++

Beijing, 2020

作者：RaySue

Code：

Agile Pioneer

文章目录

螺旋矩阵

螺旋矩阵就是顺时针螺旋排序的矩阵，由此延伸出两道题目，第一道是按照螺旋矩阵的顺序得到一维数组；第二道是把一维数组顺时针螺旋排布变为螺旋矩阵。

对于螺旋矩阵而言比较重要的其实是对于转角点的判断，往往这里的判断比较复杂，如果思路不清晰动辄百行代码，依旧bug不断，所以这个题目的正确解法是利用上下左右四指针的方法来处理。ps:如果你每次submit都需要填补丁的话，说明你的方法有问题。

题目描述

给定一个包含 m x n 个元素的矩阵（m 行, n 列），请按照顺时针螺旋顺序，返回矩阵中的所有元素。

示例 1:

输入:
[
 [ 1, 2, 3 ],
 [ 4, 5, 6 ],
 [ 7, 8, 9 ]
]
输出: [1,2,3,6,9,8,7,4,5]

示例 2:

输入:
[
  [1, 2, 3, 4],
  [5, 6, 7, 8],
  [9,10,11,12]
]
输出: [1,2,3,4,8,12,11,10,9,5,6,7]

Code

vector<int> spiralOrder(vector <vector<int>> &matrix)
{
   vector<int> ans;
   if (matrix.empty() || matrix[0].empty()) return ans;
   int m = matrix.size(), n = matrix[0].size();
   int top = 0, right = n - 1, bottom = m - 1, left = 0;
   int target = n * m;
   for(;;)
   {
       // top
       for(int i = left; i <= right; ++i) ans.push_back(matrix[top][i]);
       if (ans.size() >= target) break;
       top ++;

       // right
       for(int i = top; i <= bottom; ++i) ans.push_back(matrix[i][right]);
       if (ans.size() >= target) break;
       right --;

       // bottom
       for(int i = right; i >= left; --i) ans.push_back(matrix[bottom][i]);
       if (ans.size() >= target) break;
       bottom --;

       // left
       for(int i = bottom; i >= top; --i) ans.push_back(matrix[i][left]);
       if (ans.size() >= target) break;
       left ++;
   }
   return ans;
}

螺旋矩阵II

题目描述

给定一个正整数 n，生成一个包含 1 到 $n^2$ 所有元素，且元素按顺时针顺序螺旋排列的正方形矩阵。

示例:

输入: 3
输出:
[
 [ 1, 2, 3 ],
 [ 8, 9, 4 ],
 [ 7, 6, 5 ]
]

Code

vector <vector<int>> generateMatrix(int n)
{
   vector <vector<int>> ans(n, vector<int>(n, 0));
   int top = 0, left = 0, right = n - 1, bottom = n - 1, count = 1;
   int target = n * n;
   while(count <= target)
   {
       for (int i = left; i <= right; ++i) {ans[top][i] = count; count++;}
       top ++;
       for (int i = top; i <= bottom; ++i) {ans[i][right] = count; count++;}
       right--;
       for (int i = right; i >= left; --i) {ans[bottom][i] = count; count++;}
       bottom--;
       for (int i = bottom; i >= top; --i) {ans[i][left] = count; count++;}
       left++;
   }
   return ans;
}