动态规划f[i][j]表示前i秒时间听了j首歌的概率,则有:
f [ i ] [ j ] = ∑ f [ i − k ] [ j − 1 ] ∗ ( 1 − p j ) k − 1 ∗ p j f[i][j]=∑f[i-k][j-1]*(1-p_j)^{k-1}*p_j f[i][j]=∑f[i−k][j−1]∗(1−pj)k−1∗pj
k枚举i秒前的每一秒,要求前i-1秒都不跳歌,且在第i秒跳歌
这个方程直接转移的时间复杂度为 O ( N T 2 ) O(NT^2) O(NT2)。
尝试进行优化。
观察可以发现,f[i][j]与f[i-1][j]的方程基本不变,考虑直接从f[i-1][j]转移到f[i][j]。
时间复杂度降到 O ( N T ) O(NT) O(NT)。
#include <bits/stdc++.h>
using namespace std;
const int N=5005;
int t[N],n,m;
double p[N],f[N][N];
signed main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%lf%d",&p[i],&t[i]);
p[i]/=100; //变成概率
}
/*初始化*/
f[0][0]=1;
double sum=0;
/*DP*/
for(int i=1;i<=n;i++){
double ans=0,tp=pow(1-p[i],t[i]);
for(int j=i;j<=m;j++){
ans+=f[i-1][j-1];
if(j-t[i]>=1) ans-=f[i-1][j-t[i]-1]*tp;
f[i][j]=ans*p[i];
if(j-t[i]>=0) f[i][j]+=f[i-1][j-t[i]]*tp;
ans*=1-p[i];
sum+=f[i][j];
}
}
printf("%.9lf",sum);
}