2024牛客暑期多校训练营10 D. Is it rated?（01背包 用误差减少dp状态）

题目

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
*/