[Codeforces498B]Name That Tune（概率期望dp）

题目描述

传送门
题意：有n首音乐，第i首被听出来的概率为pi，刚开始听第一首，1s后如果听出来了则放第下一首，否则接着听这一首，第i首在连续听了ti s之后一定会被听出来，问Ts后听出来的歌的期望数量。

题解

刚开始写的方法死wa死wa…
后来发现问题了之后从i-1->i的的方法就变得死麻烦死麻烦…
终于改姿势了…

首先考虑没有ti的限制
令f(i,j)表示第j秒在听第i首歌的概率，也就是说在第j-1秒已经分辨出来第i-1首歌
那么dp方程：f(i,j+1)+=f(i,j)*(1-p(i)),f(i+1,j+1)+=f(i,j)*p(i)

但是如果有ti的限制的话，可以发现f(i,j)会向f(i,j+ti)转移，这个时候，f(i,j+ti)概率算多了，f(i+1,j+ti)概率算少了
f(i,j)有(1-p(i))^ti的概率转移到f(i,j+ti)，所以在f(i,j+ti)和f(i+1,j+ti)分别减去和加上这个偏差的部分
注意这里在调整的过程中f是不能变的，也就是要“错的调整错的”

最后答案就是sigma(i=1..n+1)f(i,T+1)*(i-1)

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
#define N 5005

int T,n,t[N];
double ans,p[N],g[N],f[N][N];

double fast_pow(double a,int p)
{
    double ans=1.0;
    for (;p;p>>=1,a*=a)
        if (p&1)
            ans*=a;
    return ans;
}
int main()
{
    scanf("%d%d",&n,&T);
    for (int i=1;i<=n;++i)
    {
        scanf("%lf%d",&p[i],&t[i]);
        p[i]/=100.0;
    }
    f[1][1]=1;p[n+1]=0;t[n+1]=5000;
    for (int i=1;i<=n+1;++i)
    {
        double P=fast_pow(1-p[i],t[i]);
        for (int j=1;j<=T+1;++j) g[j]=f[i][j];
        for (int j=1;j<=T+1;++j)
        {
            f[i][j+1]+=f[i][j]*(1-p[i]);
            f[i+1][j+1]+=f[i][j]*p[i];
            if (j+t[i]<=T+1)
            {
                f[i][j+t[i]]-=g[j]*P;
                f[i+1][j+t[i]]+=g[j]*P;
            }
        }
    }
    for(int i=1;i<=n+1;++i) ans+=f[i][T+1]*(i-1);
    printf("%.9lf\n",ans);
}