Codeforces Round 959 sponsored by NEAR (Div. 1 + Div. 2) C. Hungry Games

#include<bits/stdc++.h>
using namespace std;
const int mod=998244353;
int dx[4]={0,0,-1,1};
int dy[4]={-1,1,0,0};
#define int long long
typedef pair<int,int> pii;

const int N=200010;

int s[N];
int a[N];
int n,k;
int dp[N];

void solve()
{
	cin>>n>>k;
	memset(s,0,sizeof s);
	memset(dp,0,sizeof dp);
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
		s[i]=s[i-1]+a[i];
	}
	int ans=0;
	for(int i=n;i>=1;i--)
	{
		int l=i-1,r=n+1;
		while(l+1<r)
		{
			int mid=l+r>>1;
			if(s[mid]-s[i-1]>k)
			{
				r=mid;
			}else l=mid;
		}
		if(r==n+1)dp[i]+=n-i+1;
		else dp[i]=dp[r+1]+r-i;
	}
	for(int i=1;i<=n;i++)ans+=dp[i];
	cout<<ans<<endl;
}
															                                                                                                  
signed main()
{ 
    int good_luck_to_you;
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
   // good_luck_to_you=1; 
    cin>>good_luck_to_you;
    while(good_luck_to_you--)
    {
       solve();
    }
    system("pause");   
}

本题大意就是找到最后g不等于0的区间个数。 

主要思路：找前缀和第一次大于k的下标idx（二分），然后我们发现idx+1的方案数相当于把idx+1当作左端点来算，然后我们就想到dp[i]代表以i为左端点的方案数。

总结：在比赛的时候找到了idx，但是因为没有想到用dp来表示，所以是一直递归下去，然后因为一些边界问题还是没有写出来。

ps:本题因为需要用到后面的状态，所以我们从后往前二分来做。

一般这种区间的问题，就是以左端点或者右端点为起点，然后在较小的时间复杂度内找到符合条件的区间数。