题解 | 小红统计区间(easy)
小红统计区间(easy)
https://www.nowcoder.com/practice/96e8db05848142808e69d04d604f2dd8
#include<bits/stdc++.h>
#include <vector>
using namespace std;
using ll = long long;
int main() {
cin.tie(nullptr)->sync_with_stdio(false);
int n;
ll k;
cin >> n >> k;
vector<int> num(n);
ll cnt = 0;
for(int i = 0;i<n;i++) {
cin >> num[i];
}
ll ans = 0;
for(int l = 0,r = 0;l < n;) {
while(r < n && cnt < k) {
cnt += num[r++];
}
//cout << l << ' ' << r << endl;
if(cnt >= k) ans += n-r+1;
cnt -= num[l++];
}
cout << ans << endl;
}
很单纯的滑动窗口,因为ai>0恒成立,所以每个区间都可以向外拓展,又因为窗口向右滑动,所以只能向右拓展,不然会重复计数
查看2道真题和解析