代码随想录算法训练营第五天| Leetcode 54.螺旋矩阵、Leetcode 904.水果成篮

Leetcode 54.螺旋矩阵

题目链接：Leetcode 54.螺旋矩阵

题目描述：给你一个 m 行 n 列的矩阵 matrix ，请按照 顺时针螺旋顺序 ，返回矩阵中的所有元素。

思路：本题和Leetcode59.螺旋矩阵II 很像，不过由于这道题的矩阵不一定是正方形，我的想法是设置最外层为边界，遍历输出，输出之后设置下一层为新的边界，直到所有元素均被输出。

PS：有没有发现这个和递归的思想很像

接下来就是使用代码模拟这个过程：

首先定义一个临时数组存放已经输出了的元素：

（1）如果矩阵为空，则返回空数组

（2）如果矩阵非空，设置上下左右四个边界，循环遍历一圈后，将下一层定义为边界

如何定义循环终点呢？

当边界交错之后，说明矩阵遍历结束，跳出循环。

代码如下：

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
        vector<int> ans;
        if(matrix.empty()) return ans; //若数组为空，直接返回答案
        //定义上下左右四个边界
        int u = 0;
        int d = matrix.size() - 1;
        int l = 0;
        int r = matrix[0].size() - 1;
        while (true)  
        {
            //分别遍历上右下左四个边
            for (int i = l; i <= r; i++) ans.push_back(matrix[u][i]);
            if (++u > d) break;
            for (int i = u; i <= d; i++) ans.push_back(matrix[i][r]);
            if (--r < l) break;
            for (int i = r; i >= l; i--) ans.push_back(matrix[d][i]);
            if (--d < u) break;
            for (int i = d; i >= u; i--) ans.push_back(matrix[i][l]);
            if (++l > r) break;
        }
        return ans;
    }
};

Leetcode 904.水果成篮

题目链接：Leetcode 904.水果成篮

题目描述：

思路：

由于选定起始采摘之后只能向右移动，并且要找到水果最大数目。这道题很容易想到用滑动窗口的思想。所谓滑动窗口，就是不断的调节子序列的起始位置和终止位置，从而得出我们要想的结果。由于这道题需要将水果种类和该水果种类的个数联系起来，C++中STL有一种结构叫哈希表

，这种结构可以便于我们快速查找成对数据，一般哈希表都是用来快速判断一个元素是否出现在集合里。

对于本道题目，我们需要做的是用两个指针模拟一个窗口，根据窗口内水果种类小于等于两个这个条件，不断向后滑动窗口，并记录最多的水果个数。

代码如下：

class Solution {
public:
    int totalFruit(vector<int>& fruits) {
        int n = fruits.size();
        unordered_map<int, int> cnt; //用哈希表将水果种类和每个水果的个数联系起来
        int left = 0, ans = 0;
        for (int right = 0; right < n; right++) {
            ++cnt[fruits[right]];
            while (cnt.size() > 2) { //水果种类超过二移动左边界
                auto it = cnt.find(fruits[left]);
                --it->second;
                //如果该类水果个数为0，则将该元素从哈希表中删除
                if (it->second == 0) {
                    cnt.erase(it);
                }
                ++left;
            }
            ans = max(ans, right - left + 1);//取水果个数最大的
        }
        return ans;
    }
};

总结：今天正好有时间做一下之前的拓展题目，不过还有一道76. 最小覆盖子串 没做完，之后有时间再去尝试。思路也是滑动窗口，但是还要考虑很多细节，先放一放。

最后，如果以上文章有错误，请在评论区或私信指出，让我们共同进步！