题目链接:dd爱框框 (nowcoder.com)
一、分析题目
同向双指针(其实也就是滑动窗口)。
- 进窗口(sum += arr[right])
- 判断(sum >= x)
- 更新结果(right-left+1 < retlen retl = left retr=right)
- 出窗口(sum -= arr[left])
二、代码
1、自己 AC 的代码
//没看题解之前AC的代码
#include <iostream>
using namespace std;const int N=1e7+10, INF=1e9+7;
int a[N];int main()
{int n, x;cin >> n >> x;for(int i=1; i<=n; i++) cin >> a[i];int sum=0;int len=INF;int i=1;int l=1, r=1;for(int j=1; j<=n; j++){sum+=a[j];while(sum>=x){if(j-i+1<len){len=j-i+1;l=i, r=j;}sum-=a[i];i++;}}cout << l << ' ' << r << endl;return 0;
}//看了题解之后AC的代码
#include <iostream>
using namespace std;const int N=1e7+10;
int a[N];int main()
{int n, x;cin >> n >> x;for(int i=1; i<=n; i++) cin >> a[i];int sum=0;int len=N;int i=1, j=1;int left=1, right=1;while(j<=n){sum+=a[j];while(sum>=x){if(j-i+1<len){len=j-i+1;left=i, right=j;}sum-=a[i++];}j++;}cout << left << ' ' << right << endl;return 0;
}
2、值得学习的代码
#include <iostream>
using namespace std;const int N = 1e7 + 10;
int arr[N];
int n, x;int main()
{cin >> n >> x;for(int i = 1; i <= n; i++) cin >> arr[i];int left = 0, right = 0, sum = 0;int retLen = N, retLeft = -1, retRight = -1;while(right <= n){sum += arr[right];while(sum >= x){if(right - left + 1 < retLen){retLeft = left;retRight = right;retLen = right - left + 1;}sum -= arr[left++];}right++;}cout << retLeft << " " << retRight << endl;return 0;
}
三、反思与改进
这道题一看完,我的第一反应是前缀和,因为题目给的公式和前缀和很像,于是就往前缀和方向去想了,最后只过了 20% 的样例,我认为是数据范围的关系(因为按照题目所给的数据范围,前缀和是会超时的)。总不可能用两层 for 循环嵌套暴力解法吧,肯定超时(果然,也是只过了 20% 的样例)。于是,大脑开始飞速运转,联想到前几天好像做到一题很类似的(刚刚去查了一下,发现考察内容基本一样:209. 长度最小的子数组 - 力扣(LeetCode),我晕),在写这道题目时,想到了思路,也知道可以减少比较次数的位置,但还是没能绕开两层 for 循环这个结构,其实滑动窗口就是在暴力解法上进行优化的。本质还是对滑动窗口掌握不熟练,还有就是总喜欢固定思维让 i 先遍历,然后 j 在 i 后面遍历,这道题目可以让 j 先遍历,然后 i 在 j 之前遍历,就可以起到 “剪枝” 的作用了。但这道题如果数组中有负数,就得采用前缀和了。