代码随想录算法训练营第二天 | 双指针、滑动窗口

最新推荐文章于 2024-10-16 19:19:52 发布

Fight___

最新推荐文章于 2024-10-16 19:19:52 发布

阅读量471

点赞数 7

文章标签：数据结构

本文链接：https://blog.csdn.net/Fight___/article/details/134741289

版权

力扣题目

花费时间：3h

力扣题目记录

977.有序数组的平方

当数组满足某些规律时，就可以使用双指针

数组其实是有序的，只不过负数平方之后可能成为最大数了。

那么数组平方的最大值就在数组的两端，不是最左边就是最右边，不可能是中间。

此时可以考虑双指针法了，i指向起始位置，j指向终止位置。

定义一个新数组result，和A数组一样的大小，让k指向result数组终止位置。

如果A[i] * A[i] < A[j] * A[j] 那么result[k--] = A[j] * A[j]; 。

如果A[i] * A[i] >= A[j] * A[j] 那么result[k--] = A[i] * A[i]; 。

class Solution {
public:
    vector<int> sortedSquares(vector<int>& A) {
        int k = A.size() - 1;
        vector<int> result(A.size(), 0);
        for (int i = 0, j = A.size() - 1; i <= j;) { // 注意这里要i <= j，因为最后要处理两个元素
            if (A[i] * A[i] < A[j] * A[j])  {
                result[k--] = A[j] * A[j];
                j--;
            }
            else {
                result[k--] = A[i] * A[i];
                i++;
            }
        }
        return result;
    }
};

时间复杂度为O(n)

209.长度最小的子数组

滑动窗口

所谓滑动窗口，就是不断的调节子序列的起始位置和终止位置，从而得出我们要想的结果。

滑动窗口也可以理解为双指针法的一种，只不过这种解法更像是一个窗口的移动，所以叫做滑动窗口更适合一些。

在本题中实现滑动窗口，主要确定如下三点：

窗口内是什么？
如何移动窗口的起始位置？
如何移动窗口的结束位置？

窗口就是满足其和 ≥ s 的长度最小的连续子数组。

窗口的起始位置如何移动：如果当前窗口的值大于s了，窗口就要向前移动了（也就是该缩小了）。

窗口的结束位置如何移动：窗口的结束位置就是遍历数组的指针，也就是for循环里的索引。

滑动窗口适用连续子序列

滑动窗口的精妙之处在于根据当前子序列和大小的情况，不断调节子序列的起始位置。从而将O(n^2)暴力解法降为O(n)。

class Solution {
public:
    int minSubArrayLen(int s, vector<int>& nums) {
        int result = INT32_MAX;
        int sum = 0; // 滑动窗口数值之和
        int i = 0; // 滑动窗口起始位置
        int subLength = 0; // 滑动窗口的长度
        for (int j = 0; j < nums.size(); j++) {
            sum += nums[j];
            // 注意这里使用while，每次更新 i（起始位置），并不断比较子序列是否符合条件
            while (sum >= s) {
                subLength = (j - i + 1); // 取子序列的长度
                result = result < subLength ? result : subLength;
                sum -= nums[i++]; // 这里体现出滑动窗口的精髓之处，不断变更i（子序列的起始位置）
            }
        }
        // 如果result没有被赋值的话，就返回0，说明没有符合条件的子序列
        return result == INT32_MAX ? 0 : result;
    }
};

时间复杂度：O(n)
空间复杂度：O(1)

904.水果成篮

连续子序列，可以使用滑动窗口

但是和上一题不同的是，这一题可能会有很多数据，或多或少，且不存在什么直接关联，所以需要一个字典将他们保存下来进行修改

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)   //判断字典中元素种类是否大于2
            {
                auto it = cnt.find(fruits[left]);
                --it->second;        //从字典中删除左窗口的元素
                if (it->second == 0)
                {
                    cnt.erase(it);
                }
                ++left;
            }
            ans = max(ans, right - left + 1);    //当种类小于两种时，记录最大步长

        }
        return ans;
    }
};

54.螺旋矩阵

该题不涉及到什么算法，主要是对边界条件的处理

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
        vector<int> 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;
    }
};