解题思路
动态规划+前缀和+二分
我们用s数组来维护a数组的前缀和
状态表示:做动态规划题目时应该用集合来表示所有状态,不重不漏表示完,f[i][0]表示从前i张牌选可以赢得的最大回合数其中不包括第i张牌,f[i][1]表示从前i张牌选可以赢得的最大回合数其中包括第i张牌
这样f[i][0],f[i][1]就不重不漏表示了前i张牌的所有状态
状态计算:f[i][0]=max(f[i-1][1],f[i-1][0])这个很好理解如果第i张牌不选那么就相当于从前i-1张选,f[i][1]怎么转移呢分两种情况:
1当a[i]在l到r之间那么单独第i张牌就可以获胜了f[i][1]=max(f[i-1][0]+1,f[i-1][1]+1);
2 当a[i]小于l时:应当从下标i往前找找到一段连续的和在l和r之间的下边为L,因为前缀和是递增的所以我们考虑用二分快速的查找,找到Lf[i][1]的状态就有f[L][0]转移过来,f[i][1]=max(f[i][1],f[L][0]+1);
这样所有状态就找到完了,显然答案是max(f[n][0],f[n][1]);补充一下状态表示不唯一状态转化不唯一只要能表示所有的状态和正确的转移出来就行。
代码
#include<iostream>
#define int long long
using namespace std;
const int N=2e5+5;
int a[N];
int s[N];
int f[N][2];
int t;
int check(int l,int r,int x)
{
if(s[r]-s[l-1]>=x)
return 1;
return 0;
}
signed main()
{
cin>>t;
while(t--)
{
int n,l,r;
cin>>n>>l>>r;
for(int i=1;i<=n;i++)
cin>>a[i];
for(int i=1;i<=n;i++)
s[i]=s[i-1]+a[i];
for(int i=1;i<=n;i++)
{
f[i][0]=max(f[i-1][1],f[i-1][0]);
if(a[i]>=l&&a[i]<=r)
f[i][1]=max(f[i-1][0]+1,f[i-1][1]+1);
else if(a[i]<l)
{
int L=1,R=i;
while(L<R)
{
int mid=L+R>>1;
if(check(mid,i,l))
R=mid;
else
L=mid+1;
}
if((s[i]-s[L-1])>=l&&(s[i]-s[L-1]<=r))
f[i][1]=max(f[i][1],f[L][0]+1);
}
}
cout<<max(f[n][0],f[n][1])<<endl;
for(int i=1;i<=n;i++)
a[i]=s[i]=0,f[i][1]=f[i][0]=0;
}
return 0;
}