概率dp

自不量力开始概率dp

spoj favorite dice //倒推

题意:
一个骰子，n面，求每一面都被扔到至少一次的期望扔的次数。

思路：
$dp[i]：已经扔到i面的期望次数$
$dp[i]=dp[i-1]\ast \frac{n-i+1}{n}+dp[i]\ast \frac{i}{n}+1$
移向得：$dp[i]=dp[i-1]+\frac{n}{n-i+1}$
$f(当前状态的期望)=\sum (f(能转移过来的状态的期望)\ast 概率)+代价$

/*   Author : Rshs
 *   Data : 2019-09-11-21.13
 */
#include<bits/stdc++.h>
using namespace std;
#define FI first
#define SE second
#define LL long long
#define MP make_pair
#define PII pair<int,int>
#define SZ(a) (int)a.size()
const double pai = acos(-1);
const LL mod = 1e9+7;
const int MX = 1e6+5;
double dp[MX];
void Main(int avg){
    int n;cin>>n;
    dp[0]=0;
    for(int i=1;i<=n;i++) dp[i]=dp[i-1]+(1.0*n)/(1.0*n-1.0*i+1.0);
    cout<<dp[n]<<'\n';
}
int main(){
    int cas;cin>>cas;for(int i=1;i<=cas;i++)Main(i);
    return 0;
}

LOOPS //二维倒推

题目

题意：
一个人从(1,1)走到(n,m)，当他在(i,j)时，可以花费2魔力施法，有p1几率原地不动，p2几率向右走，p3几率向下走。求到(n,m)的期望花费的魔力。
思路：

$正着不好推，考虑倒推$。
$dp[i][j]:从(i,j)到(n,m)的期望，dp[n][m]=0$
$dp[i][j]=2+dp[i][j]\ast p1[i][j]+dp[i+1][j]\ast p2[i][j]+dp[i][j+1]\ast p3[i][j]$
$移向得：dp[i][j]=(2+dp[i+1][j]\ast p2[i][j]+dp[i][j+1]\ast p3[i][j])/(1-p1[i][j])$
$这里注意p1[i][j]==1时候,直接continue.$$保证有解,说明走到这种点的概率为0$

/*   Author : Rshs
 *   Data : 2019-09-12-15.15
 */
#include<bits/stdc++.h>
using namespace std;
#define FI first
#define SE second
#define LL long long
#define MP make_pair
#define PII pair<int,int>
#define SZ(a) (int)a.size()
const double pai = acos(-1);
const double eps = 1e-7;
const LL mod = 1e9+7;
const int MX = 1e3+5;
double p1[MX][MX],p2[MX][MX],p3[MX][MX],dp[MX][MX];

int main(){
    int n,m;
    while(cin>>n>>m){
        for(int i=1;i<=n;i++)for(int j=1;j<=m;j++)
            scanf("%lf%lf%lf",&p1[i][j],&p2[i][j],&p3[i][j]);
        dp[n][m]=0;
        for(int i=n;i>=1;i--)for(int j=m;j>=1;j--){
            if(i==n&&j==m)continue;
            if(fabs(p1[i][j]-1)<=eps) {dp[i][j]=1e8;continue;} //不会走到这种点，但是需要continue
            dp[i][j]=(2.0+dp[i][j+1]*p2[i][j]+dp[i+1][j]*p3[i][j])/(1.0-p1[i][j]);
        }
        printf("%.3lf\n",dp[1][1]);
    }
    return 0;
}

Help Me Escape //倒推

题目

$题意：$
$ly初始战斗力为f，ly周围有n条路，每条路上有一个妖怪，战斗力为c[i]。$
$每天，ly等概率的选一条路。$
$如果他战斗力大于c[i]，那么他花t[i]天就可以逃出去。$
$否则他战斗力变为f+c[i]，等明天再战。$
$求ly逃出去的天数的期望$

$思路：$
$考虑倒推，因为如果大于所有的妖怪，期望直接可以求出来$
$dp[i]：战斗力为i时，逃出去还需要的期望天数$

$dp[i]={\sum }_{j=1}^n\frac{i>c[j]?t[j]:(dp[i+c[j]]+1)}{n}$
$dp[inf]={\sum }_{j=1}^n\frac{t[j]}{n}$

/*   Author : Rshs
 *   Data : 2019-09-12-16.00
 */
#include<bits/stdc++.h>
using namespace std;
#define FI first
#define SE second
#define LL long long
#define MP make_pair
#define PII pair<int,int>
#define SZ(a) (int)a.size()
const double pai = acos(-1);
const LL mod = 1e9+7;
const int MX = 1e6+5;
int c[105];
double t[105],dp[MX];
int main(){
    int n,f;
    while(cin>>n>>f){
       int mx=2e4;
        for(int i=1;i<=n;i++){
            cin>>c[i];
            t[i]=(int)(1.0*c[i]*c[i]*(1.0+sqrt(5))/2);
        }
        for(int i=mx;i>=f;i--){
            dp[i]=0;
            for(int j=1;j<=n;j++){
                if(i>c[j]){
                    dp[i]=dp[i]+t[j]/(double)n; //成功
                }
                else {
                    dp[i]=dp[i]+(1.0+dp[i+c[j]])/(double)n; //失败
                }
            }
        }
        printf("%.3lf\n",dp[f]);
    }
    return 0;
}

hdu4035 Maze //循环解方程

题目

题意：
$n$个节点的树，从结点1出发，到每个结点$i$有三个事件：
$1.k_i$概率被杀死回到节点$1$
$2.e_i$概率找到节点上面的出口逃出并胜利
$3.(1-k_i-e_i)$概率下等概率走一条相连的边
求胜利所要走的步数的期望

思路:

$神仙题目了.......$

$f(i):在i点时逃出所需要的期望次数$

$叶子节点时：$
$f(i)=e_i\ast 0+k_i\ast f(1)+(1-e_i-k_i)\ast (f(fa[i])+1)$
$f(i)=k_i\ast f(1)+(1-e_i-k_i)\ast f(fa[i])+1-e_i-k_i$
$非叶子节点：$
$f(i)=e_i\ast 0+k_i\ast f(1)+\frac{(1-e_i-k_i)}{d_i}\ast (f(fa[i])+1)+\frac{(1-e_i-k_i)}{d_i}\ast ({\sum }_{j\in son(i)}(f(j)+1))$
$f(i)=k_i\ast f(1)+\frac{(1-e_i-k_i)}{d_i}\ast f(fa[i])+\frac{(1-e_i-k_i)}{d_i}\ast ({\sum }_{j\in son(i)}f(j))+1-e_i-k_i①$

$设f(i)=a_i\ast f(1)+b_i\ast f(fa[i])+c_i.$

$对于①式:$
$f(i)=k_i\ast f(1)+\frac{(1-e_i-k_i)}{d_i}\ast f(fa[i])+\frac{(1-e_i-k_i)}{d_i}\ast ({\sum }_{j\in son(i)}(a_j\ast f(1)+b_j\ast f(fa[j])+c_j))+1-e_i-k_i$
$因为fa[j]=i:$
$f(i)=k_i\ast f(1)+\frac{(1-e_i-k_i)}{d_i}\ast f(fa[i])+\frac{(1-e_i-k_i)}{d_i}\ast ({\sum }_{j\in son(i)}(a_j\ast f(1)+b_j\ast f(i)+c_j))+1-e_i-k_i$
$把右边的f(i)移到左边：$
$f(i)\ast (1-\frac{(1-e_i-k_i)}{d_i}\ast ({\sum }_{j\in son(i)}{b}_j))=f(1)\ast (ki+\frac{(1-e_i-k_i)}{d_i}\ast ({\sum }_{j\in son(i)}{a}_j))+\frac{(1-e_i-k_i)}{d_i}\ast f(fa[i])+1-e_i-k_i+\frac{(1-e_i-k_i)}{d_i}\ast ({\sum }_{j\in son(i)}{c}_j)$

$可得：$

$$a_i=\frac{k_i+\frac{(1-e_i-k_i)}{d_i}\ast ({\sum }_{j\in son(i)}{a}_j)}{1-\frac{(1-e_i-k_i)}{d_i}\ast ({\sum }_{j\in son(i)}{b}_j)}$$
$$b_i=\frac{\frac{(1-e_i-k_i)}{d_i}}{1-\frac{(1-e_i-k_i)}{d_i}\ast ({\sum }_{j\in son(i)}{b}_j)}$$
$$c_i=\frac{1-e_i-k_i+\frac{(1-e_i-k_i)}{d_i}\ast ({\sum }_{j\in son(i)}{c}_j)}{1-\frac{(1-e_i-k_i)}{d_i}\ast ({\sum }_{j\in son(i)}{b}_j)}$$

$这时候一个神奇的地方，对于叶子节点$：
$a_i=k_i,b_i=1-e_i-k_i,c_i=1-e_i-k_i$
$那么从叶子节点开始求，不停递归回根节点.$
$\because f(1)=a_1\ast f(1)+c_1\therefore f(1)=\frac{c_1}{1-a_1}$
$这里判断无解的条件：a_1趋近于1$

/*   Author : Rshs
 *   Data : 2019-09-12-19.20
 */
#include<bits/stdc++.h>
using namespace std;
#define FI first
#define SE second
#define LL long long
#define MP make_pair
#define PII pair<int,int>
#define SZ(a) (int)a.size()
const double pai = acos(-1);
const double eps = 1e-9;
const LL mod = 1e9+7;
const int MX = 1e5+5;
double e[MX],k[MX];
double a[MX],b[MX],c[MX],aa[MX],bb[MX],cc[MX];
vector<int>g[MX];
void dfs(int v,int fa){
    aa[v]=bb[v]=cc[v]=0;
    for(auto i:g[v]){
        if(i==fa) continue;
        dfs(i,v);
        aa[v]+=a[i];bb[v]+=b[i];cc[v]+=c[i];
    }
    double zz=(1.0-e[v]-k[v])/(1.0*SZ(g[v]));
    a[v]=(k[v]+zz*aa[v])/(1.00-zz*bb[v]);
    b[v]=zz/(1.0-zz*bb[v]);
    c[v]=(1.0-e[v]-k[v]+zz*cc[v])/(1.0-zz*bb[v]);
}
void Main(int avg){
    int n;cin>>n;
    for(int i=1;i<=n;i++)g[i].clear();
    for(int i=1;i<n;i++){
        int sa,sb;scanf("%d%d",&sa,&sb);
        g[sa].push_back(sb);g[sb].push_back(sa);
    }
    for(int i=1;i<=n;i++)scanf("%lf%lf",&k[i],&e[i]),k[i]/=100.0,e[i]/=100.0;
    dfs(1,-1);
    printf("Case %d: ",avg);
    if(fabs(a[1]-1)<eps) {puts("impossible");return;}
    printf("%lf\n",c[1]/(1-a[1]));
}
int main(){
    int cas;cin>>cas;for(int i=1;i<=cas;i++)Main(i);
    return 0;
}

Bloodsucker //组合数

题目
题意：
$n$个人，一开始有一个吸血鬼，每天随机的从$n$个人中挑出两个人，如果一个人是吸血鬼，一个人是普通人，那么普通人有$p$的概率被感染，求全部人变成吸血鬼的期望天数。
思路：
$定义dp[i]:有i个吸血鬼时，全部变成吸血鬼的期望天数，dp[n]=0.$
$第i天感染一个人的概率为：z=\frac{i\ast (n-i)}{n\ast (n-1)/2}.$
$dp[i]=(1+dp[i+1])\ast z+(dp[i]+1)\ast (1-z),然后移项即可$

/*   Author : Rshs
 *   Data : 2019-09-13-15.27
 */
#include<bits/stdc++.h>
using namespace std;
#define FI first
#define SE second
#define LL long long
#define MP make_pair
#define PII pair<int,int>
#define SZ(a) (int)a.size()
const double pai = acos(-1);
const double eps = 1e-10;
const LL mod = 1e9+7;
const int MX = 1e6+5;
double dp[MX];
void Main(int avg){
    int n;double p;
    cin>>n>>p;
    dp[n]=0;
    for(int i=n-1;i>=1;i--){
        double x=(1.0*n-i)*2/(1.0*n);
        double y=1.0*i/(1.0*n-1);
        dp[i]=(1.0+dp[i+1]*x*y*p)/(x*y*p);
    }
    printf("%.3lf\n",dp[1]);
}
int main(){
    int cas;cin>>cas;for(int i=1;i<=cas;i++)Main(i);
    return 0;
}

CF #105 Div.2 D. Bag of mice//概率,二维

题目
题意：
有$w$个白老鼠，$b$个黑老鼠，$ly$和$rs$博弈，先取到白老鼠的人获胜，$ly$先手。
当$rs$取的时候，会随机的跑掉一只老鼠。没人取到白老鼠$rs$获胜。
求$ly$获胜的期望.
思路：
定义$dp[i][j][k]$
当$k==0$时：剩下$i$个白老鼠，$j$个黑老鼠，轮到$ly$取老鼠的概率。
当$k==1$时：剩下$i$个白老鼠，$j$个黑老鼠，轮到$rs$取老鼠的概率。
最后从$(w,b)$倒推矩阵到$(0,0)$，期间累加ly获胜情况下的概率。

/*   Author : Rshs
 *   Data : 2019-09-13-20.23
 */
#include<bits/stdc++.h>
using namespace std;
#define FI first
#define SE second
#define LL long long
#define MP make_pair
#define PII pair<int,int>
#define SZ(a) (int)a.size()
const double pai = acos(-1);
const double eps = 1e-10;
const LL mod = 1e9+7;
const int MX = 1e6+5;
double dp[1005][1005][2];
int main(){
    int w,b;
    cin>>w>>b;dp[w][b][0]=1;
    double ans=0;
    for(int i=w;i>=0;i--){
        for(int j=b;j>=0;j--){
            if(i==0&&j==0) continue;
            double p=(i*1.00)/(1.0*i+j);
            ans+=p*dp[i][j][0];
            if(j>=1)dp[i][j-1][1]+=(1.0-p)*dp[i][j][0];
            if(j>=2)dp[i][j-2][0]+=(1.0-p)*(1.0*j-1)/(1.0*j-1+i)*dp[i][j][1];
            if(j>=1&&i>=1) dp[i-1][j-1][0]+=(1.0-p)*(1.0*i)/(1.0*j-1+i)*dp[i][j][1];
        }
    }
    printf("%.12lf\n",ans);
    return 0;
}