CF 1994 C. Hungry Games (*1600) 思维+二分
题意:
给你一个长度为 \(n\) 的关卡,和一个正整数 \(x\),初始分数为 \(0\),通过每个关卡就会获得对应的分数。
但是分数如果超过 \(x\),就会清零。现在让你求出满足最终得分不为零的所有子区间数量。
思路:
正难则反,改求最终得分为零的所有子区间数量。考虑维护前缀和数组,然后枚举左端点。
每次在前缀和数组中查询对应的右端点即可。
代码:
#include<bits/stdc++.h>
using namespace std;
#define ff first
#define ss second
#define pb push_back
#define all(u) u.begin(), u.end()
#define endl '\n'
#define debug(x) cout<<#x<<":"<<x<<endl;
typedef pair<int, int> PII;
typedef long long LL;
const int inf = 0x3f3f3f3f;
const int N = 1e5 + 10, M = 105;
const int mod = 1e9 + 7;
const int cases = 1;
void Showball(){
int n,x;
cin>>n>>x;
vector<int> a(n);
vector<LL> s(n+1);
for(int i=0;i<n;i++){
cin>>a[i];
s[i+1]=s[i]+a[i];
}
LL ans=1LL*n*(n+1)/2;
vector<int> f(n+1);
for(int i=n-1;i>=0;i--){
auto p=lower_bound(all(s),s[i]+x+1)-s.begin();
f[i]=(p==n+1?0:f[p]+1);
ans-=f[i];
}
cout<<ans<<endl;
}
int main(){
ios::sync_with_stdio(false);
cin.tie(nullptr);
cout.tie(nullptr);
int T=1;
if(cases) cin>>T;
while(T--)
Showball();
return 0;
}
标签:const,1994,int,1600,CF,Hungry,vector,define
From: https://www.cnblogs.com/showball/p/18393733