题目大意:
已知一个数列,数字都为正数 若里面的一个连续子串大于K,我们则认为这个子串是满足要求的子串,问这个数列有多少个这种满足要求的子串。
解题思路:
一开始以为是区间dp,后来因为我比较辣鸡推不出,所以换了个naive的思路,结果就可以了。
很直接的,我们有一个N^2的思路,每次枚举数列的一个数作为这个子串的起点,然后从这里开始出发区域求和到J点,若和大于K,那么N-J即为这个起点的满足要求的所有串的数(0=<J <= N - 1 , N为串的长度),然后所有起点的串的数 加起来即为结果,可惜这方法会TLE.
但是我们可以注意到,一个区域求和满足要求,那么我们的起点往下走的同时,我们其实不需要重新算一遍同样的区间和!只需把上一次的和减去上一次的区间的起点,然后再判断是否满足大于K,不满足再拓展求和区域即可!
AC代码:
#include <bits/stdc++.h>
using namespace std;
int main() {
long long n, k;
cin >> n >> k;
vector<long long> mv;
for (int i = 0; i<n; i++) {
long long t;
cin >> t;
mv.push_back(t);
}
long long sum = 0;
long long arr_c = 0;
int j = 0;
int suc_flag = 0;
long long sz = (long long)mv.size();
for (long long i = 0; i<(long long)mv.size(); i++) {
if(i>=j){
sum = mv[i];
j = i;
}
else sum -= mv[i - 1];
while(1){
if (sum >= k){
arr_c += n - j;
break;
}
else{
j++;
if (j>sz - 1){
suc_flag = 1;
break;
}
sum += mv[j];
}
}
if (suc_flag)break;
}
cout << arr_c << endl;
return 0;
}