zoj3329 One Person Game（期望dp）

关键词：期望dp，待定系数法
题意：已知有三个骰子，分别有k1,k2,k3个面。一开始cnt=0.。当骰子掷为a,b,c时，cnt归零，否则将三个数的和加入cnt。求第一次cnt>n的时，掷筛子的期望次数。
dp[i]:cnt=i时离cnt>n状态还需要掷的期望次数
$dp[i]=(dp[0]+\sum_{x=1}^{k1}\sum_{y=1}^{k2}\sum_{z=1}^{k3}_{(x,y,z)!=(a,b,c)}dp[i+x+y+z])+1$
式中含dp[0]，如何去除dp[0](最终要求的状态——待定系数法
设通式dp[i]=A[i]*dp[0]+B[i]，代入上面的递推式，对应各项可分别得到A[i]和B[i]的递推式（不含dp[0]）
求出A[0],B[0]，dp[0]=A[0]*dp[0]+B[0]，求出dp[0]即可！

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<set>
#include<vector>
#include<map>
#define ll long long
using namespace std;

const int maxn = 500+10;

int t;
int n,k1,k2,k3,a,b,c;
double dp[maxn],A[maxn],B[maxn];

int main(){
    scanf("%d",&t);
    while(t--){
        memset(dp,0,sizeof(dp));
        memset(A,0,sizeof(A)),memset(B,0,sizeof(B));
        scanf("%d%d%d%d%d%d%d",&n,&k1,&k2,&k3,&a,&b,&c);
        for(int i=n;i>=0;i--){
            double tmp1=0.0,tmp2=0.0;
            for(int x=1;x<=k1;x++){
                for(int y=1;y<=k2;y++){
                    for(int z=1;z<=k3;z++){
                        if(x==a&&y==b&&z==c) continue;
                        tmp1+=A[i+x+y+z];
                        tmp2+=B[i+x+y+z];
                    }
                }
            }
            A[i]=(double)(1+tmp1)/(double)(k1*k2*k3);
            B[i]=(double)(1+(tmp2/(double)(k1*k2*k3)));
        }
        printf("%.10lf\n",B[0]/(1-A[0]));
    }
    return 0;
}