ZOJ 3582 Back to the Past（概率DP）

题目链接：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=4624

题意：A和B上各有n盏灯，初始时都是灭的。已知每天每个灯亮的概率为p。且亮了以后就不会再灭了。求A和B上各有不少于m盏灯亮的天数期望。

思路：f[i][j]表示分别还有i个和j个是灭的期望。

import java.util.*;

import java.text.*;

import java.math.*;



public class Main{

    

    static double EPS=1e-10;

    static double PI=Math.acos(-1.0);

    

    static double f[][]=new double[55][55];

    static double C[][]=new double[55][55];

    static double Log[]=new double[105];

    static double Log1[]=new double[105];

    static int n,m;

    static double p;

    

    

    public static void PR(String s){

        System.out.println(s);

    }

    

    public static void PR(int x)

    {

        System.out.println(x);

    }

    

    public static void PR(double s)

    {

        java.text.DecimalFormat d=new java.text.DecimalFormat("#.000000");

        System.out.println(d.format(s));

    }

    

    public static double DFS(int n1,int n2)

    {

        if(f[n1][n2]>=0) return f[n1][n2];

        if(n-n1>=m&&n-n2>=m) return 0;

        double ans=1,x;

        int i,j;

        for(i=0;i<=n1;i++) for(j=0;j<=n2;j++)

        {

            if(i==0&&j==0) continue;

            ans+=DFS(n1-i,n2-j)*C[n1][i]*Log[i]*Log1[n1-i]*C[n2][j]*Log[j]*Log1[n2-j];

        }

        ans/=(1-Log1[n1]*Log1[n2]);

        f[n1][n2]=ans;

        return ans;

    }

    

    public static void init()

    {

        int i,j;

        for(i=0;i<=50;i++)

        {

            C[i][0]=C[i][i]=1;

            for(j=1;j<i;j++) C[i][j]=C[i-1][j-1]+C[i-1][j];

        }

    }

    

    public static void main(String[] args){

  

        init();

        Scanner S=new Scanner(System.in);

        while(S.hasNext())

        {

            n=S.nextInt();

            m=S.nextInt();

            p=S.nextDouble();

            if(n==0&&m==0&&p==0) break;

            int i,j;

            for(i=0;i<=n;i++) for(j=0;j<=n;j++) 

            {

                f[i][j]=-1;

            }

            Log[0]=Log1[0]=1;

            for(i=1;i<=n;i++) Log[i]=Log[i-1]*p;

            for(i=1;i<=n;i++) Log1[i]=Log1[i-1]*(1-p);

            PR(DFS(n,n));

        }

    }

}