题目
t(t<=1000)组样例,每次给定n(n<=1e5)场比赛,
有一个浮点系数k(0.1<=k<=1),对于n场来说系数是一样的,
第i场还有一个rating变量pi,假设你参加这场比赛,
初始rating是r,那么比赛结束后会变成k*pi+(1-k)*r
你可以skip掉其中不超过m(m<=n)场,给定初始rating为r0,
问最大的终态浮点数rating是多少,误差不超过1e-9视为正确
思路来源
jiangly代码
题解
考虑参加两场比赛时,这个式子的合并,
r=k*pi+(1-k)*r
r=(k*pi+(1-k)*(k*pj+(1-k)*r0))
=k*pi+(1-k)*k*pj+(1-k)^2*r0
倒着看这个比赛的过程,
倒着参加的最后一场(不妨第i场)的贡献固定是k*pi,
倒数第二场(不妨第j场)固定是k*(1-k)*pj,
r0后面参加了x场,r0的贡献就是r0*(1-k)^x
所以可以倒着dp这个过程,就无后效了,dp[i]表示当前参加了i场时的最大rating和,
只要参加的总场数不小于n-m即可
然后这么暴力dp是O(n^2)的,不能接受,注意到k>=0.1,1-k<=0.9
那么(1-k)的x次方,当x到一定次数就很小了,试着400多次方的时候就会小于1e-19,
此时乘以pi=1e5也小于1e-14,乘以1e5场也不会超过1e-9,对答案的误差不会产生影响
所以,为了场次满足要求,在选够M=400场时,可以直接把后面贡献≈0的场直接选上
所以只在i=0或j=m时更新答案,i=0是考虑可能n很小的情况
代码
#include <bits/stdc++.h>
using namespace std;
#define rep(i,a,b) for(int i=(a);i<=(b);++i)
#define per(i,a,b) for(int i=(a);i>=(b);--i)
typedef long long ll;
typedef double db;
typedef pair<int,int> P;
#define fi first
#define se second
#define pb push_back
#define dbg(x) cerr<<(#x)<<":"<<x<<" ";
#define dbg2(x) cerr<<(#x)<<":"<<x<<endl;
#define SZ(a) (int)(a.size())
#define sci(a) scanf("%d",&(a))
#define pt(a) printf("%d",a);
#define pte(a) printf("%d\n",a)
#define ptlle(a) printf("%lld\n",a)
#define debug(...) fprintf(stderr, __VA_ARGS__)
using namespace std;
typedef long long ll;
const int N=1e5+10,M=400;
int t,n,m,a[N];
db k,r0,p[N],dp[M+5];
int main(){
std::ios::sync_with_stdio(false);
std::cin.tie(nullptr);
cin>>t;
while(t--){
cin>>n>>m>>k;
p[0]=1;
rep(i,1,M)p[i]=p[i-1]*(1-k);
cin>>r0;
memset(dp,0,sizeof dp);
rep(i,0,n-1)cin>>a[i];
db ans=0;
per(i,n-1,0){
per(j,M,0){
if(j)dp[j]=max(dp[j],dp[j-1]+k*a[i]*p[j-1]);
if(i==0 || j==M){
if(n-i-j<=m)ans=max(ans,dp[j]+r0*p[j]);
}
}
}
cout<<fixed<<setprecision(10)<<ans<<endl;
}
return 0;
}
/*
r=kpi+(1-k)*r
r=(kpi+(1-k)*(kpj+(1-k)*r0))
=kpi+(1-k)*k*pj+(1-k)^2*r
*/