代码随想录第二天:滑动窗口、前缀和

1.最小长度子数组

题目链接:. - 力扣(LeetCode)

文章讲解:代码随想录

视频讲解:拿下滑动窗口! | LeetCode 209 长度最小的子数组_哔哩哔哩_bilibili

本题使用滑动窗口思想,要注意子数组在原数组中是连续的。

定义左右窗口指针,统计窗口内部数字之和。

如果和小于目标值,就让右指针向右移动,和加上右指针的值。

如果和大于等于目标值,就让和减去左指针的值,再让左指针向右移动一格。

终止条件是右指针超出数组范围。

代码如下:

    int minSubArrayLen(int target, vector<int> &nums) {
        int i = 0;
        int result = INT_MAX;
        int count = 0;
        int length = 0;
        for (int j = 0; j < nums.size(); ++j) {
            count += nums[j];
            while (count >= target) {
                length = j - i + 1;
                result = std::min(result, length);
                count -= nums[i++];
            }
        }
        return result == INT_MAX ? 0 : result;
    }

2.螺旋矩阵

题目链接:. - 力扣(LeetCode)

文章讲解:代码随想录

视频讲解:一入循环深似海 | LeetCode:59.螺旋矩阵II_哔哩哔哩_bilibili

本题主要考察逻辑能力

要将遍历过程分成若干循环:向右,向下,向左,向上。如此循环。

可以将每个操作使用for循环实现,要注意边界条件,即边角位置只能被一个for循环使用!

注意每次循环的开始位置不同,所以每次循环开始前定义起始位置。

注意每次循环四次操作的结束位置不同,所以用变量定义结束位置。

注意设置循环条件不应包括单数矩阵的中心点,中心点要在循环外额外定义。

代码如下:

vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> res(n, vector<int>(n, 0)); // 使用vector定义一个二维数组
        int startx = 0, starty = 0; // 定义每循环一个圈的起始位置
        int loop = n / 2; // 每个圈循环几次,例如n为奇数3,那么loop = 1 只是循环一圈,矩阵中间的值需要单独处理
        int mid = n / 2; // 矩阵中间的位置,例如:n为3, 中间的位置就是(1,1),n为5,中间位置为(2, 2)
        int count = 1; // 用来给矩阵中每一个空格赋值
        int offset = 1; // 需要控制每一条边遍历的长度,每次循环右边界收缩一位
        int i, j;
        while (loop--) {
            i = startx;
            j = starty;
            // 下面开始的四个for就是模拟转了一圈
            // 模拟填充上行从左到右(左闭右开)
            for (j; j < n - offset; j++) {
                res[i][j] = count++;
            }
            // 模拟填充右列从上到下(左闭右开)
            for (i; i < n - offset; i++) {
                res[i][j] = count++;
            }
            // 模拟填充下行从右到左(左闭右开)
            for (; j > starty; j--) {
                res[i][j] = count++;
            }
            // 模拟填充左列从下到上(左闭右开)
            for (; i > startx; i--) {
                res[i][j] = count++;
            }
            // 第二圈开始的时候,起始位置要各自加1, 例如:第一圈起始位置是(0, 0),第二圈起始位置是(1, 1)
            startx++;
            starty++;
            // offset 控制每一圈里每一条边遍历的长度
            offset += 1;
        }
        // 如果n为奇数的话,需要单独给矩阵最中间的位置赋值
        if (n % 2) {
            res[mid][mid] = count;
        }

3.区间和

文章讲解:58. 区间和 | 代码随想录

本题的重点在于查找区间和的时间,如果数据量非常大,使用常规for循环累加的方法的时间复杂度会达到O(n * m),n是数据个数,m是查找次数。

前缀和数组p将原数组下标前缀为n及其前面所有元素相加作为counts[n],而每次查找区间和(a,b)时只需要使用前缀和数组counts[b]-counts[a-1]就能得到答案(a=0时特殊对待),每次查找的时间复杂度为O(1)。

代码如下:

#include <iostream>
#include <vector>

using namespace std;

int main() {
    int n;
    cin >> n;
    int temp = 0;
    int a, b;
    vector<int> nums(n);
    vector<int> counts(n);
    for (int i = 0; i < n; ++i) {
        cin >> nums[i];
        temp += nums[i];
        counts[i] = temp;
    }
    while (cin >> a >> b){
        int sum;
        if (a == 0)
            sum = counts[b];
        else
            sum = counts[b] - counts[a-1];
        cout << sum << endl;
    }
    return 0;
}

4.开发商购买土地

文章讲解:44. 开发商购买土地 | 代码随想录 (programmercarl.com)

本题的意思是一个二维矩阵,只能横着或者竖着切一刀,切完两边元素之和的差最小是多少。因此我们需要知道每次切完两边元素和的差。

这其实是一个二维前缀和的问题。对待数据量大的数据,使用for循环累加同样时间复杂度很高,这里可以计算两个前缀和数组,分别计算横向和竖向的累计和。这样我们就可以得到横向和竖向任意区间和,进而解决切完之后的元素和问题。

代码如下:

#include <iostream>
#include <vector>
#include <climits>

using namespace std;
int main () {
    int n, m;
    cin >> n >> m;
    int sum = 0;
    vector<vector<int>> vec(n, vector<int>(m, 0)) ;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            cin >> vec[i][j];
            sum += vec[i][j];
        }
    }
    // 统计横向
    vector<int> horizontal(n, 0);
    for (int i = 0; i < n; i++) {
        for (int j = 0 ; j < m; j++) {
            horizontal[i] += vec[i][j];
        }
    }
    // 统计纵向
    vector<int> vertical(m , 0);
    for (int j = 0; j < m; j++) {
        for (int i = 0 ; i < n; i++) {
            vertical[j] += vec[i][j];
        }
    }
    int result = INT_MAX;
    int horizontalCut = 0;
    for (int i = 0 ; i < n; i++) {
        horizontalCut += horizontal[i];
        result = min(result, abs(sum - horizontalCut - horizontalCut));
    }
    int verticalCut = 0;
    for (int j = 0; j < m; j++) {
        verticalCut += vertical[j];
        result = min(result, abs(sum - verticalCut - verticalCut));
    }
    cout << result << endl;
}

第二十二天的算法训练营主要涵盖了Leetcode题目中的三道题目,分别是Leetcode 28 "Find the Index of the First Occurrence in a String",Leetcode 977 "有序数组的平方",和Leetcode 209 "长度最小的子数组"。 首先是Leetcode 28题,题目要求在给定的字符串中找到第一个出现的字符的索引。思路是使用双指针来遍历字符串,一个指向字符串的开头,另一个指向字符串的结尾。通过比较两个指针所指向的字符是否相等来判断是否找到了第一个出现的字符。具体实现的代码如下: ```python def findIndex(self, s: str) -> int: left = 0 right = len(s) - 1 while left <= right: if s[left == s[right]: return left left += 1 right -= 1 return -1 ``` 接下来是Leetcode 977题,题目要求对给定的有序数组中的元素进行平方,并按照非递减的顺序返回结果。这里由于数组已经是有序的,所以可以使用双指针的方法来解决问题。一个指针指向数组的开头,另一个指针指向数组的末尾。通过比较两个指针所指向的元素的绝对值的大小来确定哪个元素的平方应该放在结果数组的末尾。具体实现的代码如下: ```python def sortedSquares(self, nums: List[int]) -> List[int]: left = 0 right = len(nums) - 1 ans = [] while left <= right: if abs(nums[left]) >= abs(nums[right]): ans.append(nums[left ** 2) left += 1 else: ans.append(nums[right ** 2) right -= 1 return ans[::-1] ``` 最后是Leetcode 209题,题目要求在给定的数组中找到长度最小的子数组,使得子数组的和大于等于给定的目标值。这里可以使用滑动窗口的方法来解决问题。使用两个指针来表示滑动窗口的左边界和右边界,通过移动指针来调整滑动窗口的大小,使得滑动窗口中的元素的和满足题目要求。具体实现的代码如下: ```python def minSubArrayLen(self, target: int, nums: List[int]) -> int: left = 0 right = 0 ans = float('inf') total = 0 while right < len(nums): total += nums[right] while total >= target: ans = min(ans, right - left + 1) total -= nums[left] left += 1 right += 1 return ans if ans != float('inf') else 0 ``` 以上就是第二十二天的算法训练营的内容。通过这些题目的练习,可以提升对双指针和滑动窗口算法的理解和应用能力。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值