解题思路:
首先,不能使用暴力枚举,时间为O(n2),超时。以下为正确做法:
假设找到一段区间(其和>=m),如上图黄色部分,那么该区间加上i后面的元素形成的新区间和都>=m,因此以该区间为基础就有n-i+1个区间符合要求。
那么我们只需要从1开始找到每一个恰好大于等于m的黄色区间,再依次把每一个黄色区间为基础的区间的个数相加就得到答案。
AC代码:
cpp
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int N = 1e5+9;
int a[N];
ll m;
// 依次找出区间和>=m的滑动窗口,j++
int main()
{
ll sum = 0,ans = 0;
int n, j = 1;cin >> n >> m;
for(int i = 1; i <= n; i++){
cin >> a[i];
sum += a[i];
if(sum >= m)
{
ans += (n-i+1);
while(j <= i && sum >= m){ // 数组从1开始序号递增,所以当序号i>=j时区间合法
sum -= a[j];
j++;
if(sum >= m)ans += (n-i+1);
}
}
}
cout << ans << '\n';
return 0;
}
知识点:
双指针,滑动窗口