题意:
给定 n x
表示 n 个数 ,(每个数大于等于1,小于等于2e5)g 表示 区间的和,当g>x 的时候,将 g 变成零。询问有多少区间满足 g 不为零
比较套路的题,因为询问的是子串,一共有n^2 中状态
一般思路:枚举左/右端点,以0(小复杂度),可以是logn 或者是 log n^2
计算有多少满足题意的右/左端点。
思路:
枚举左端点l ,看有多少满足题意的r 。
首先我们要找到,第一个使 sum 清零的点。因为sum 是单增的,所以我们可以二分出来,第一个>x 的点,记为k;
满足条件的r ,从 l 到 k-1 ,都是满足条件的 k 位的时候被清零了额。
当k+1 位 的时候 ,相当于 l=k+1
递推过来
表示以i 为左端点 ,满足 情况的方案数。
i n ->1。对于每一个i 二分出来k (区间和 用 前缀和来 维护)
f[i]= k-i+dp[k+1]
当然还有一些边界的问题。比如 k 是 数组的最后一位,
还有 找不到k 的情况。
#include <bits/stdc++.h>
using namespace std;
#define int long long
void solve()
{
int n,x; cin>>n>>x;
vector<int>a(n+1,0);
vector<int>dp(n+2,0);//数组多开一格,这样 k ==n 的时候
// 不用额外处理了
for(int i=1;i<=n;i++)
{
cin>>a[i];a[i]+=a[i-1];
}
for (int i=n;i>=1;i--)
{
int l=i,r=n;bool flag=false;
while(l<=r)
{
//枚举 这个点 ,以i 为左端点 sum值是零
int mid= l+r >>1;
if (a[mid]-a[i-1]>x){
r=mid-1;
flag=true;
}
else l=mid+1;
}
if (!flag)dp[i]=n-i+1;
else {
dp[i]=r+1-i+dp[r+1+1];
}
}
int ans=0;
for (int i=1;i<=n;i++)
{
ans+=dp[i];
}
cout<<ans<<"\n";
}
signed main()
{
std::cin.tie(nullptr)->sync_with_stdio(false);
int t;cin>>t;
while(t--)
{
solve();
}
return 0;
}