【BZOJ 4832 】 4832: [Lydsy2017年4月月赛]抵制克苏恩 （期望DP）

 1 #include<cstdio>
 2 #include<cstdlib>
 3 #include<cstring>
 4 #include<iostream>
 5 #include<algorithm>
 6 using namespace std;
 7 
 8 bool vis[60][10][10][10];
 9 double f[60][10][10][10];
10 
11 double ffind(int k,int a,int b,int c)
12 {
13     if(k==0||a<0||b<0||c<0) return 0;
14     if(vis[k][a][b][c]) return f[k][a][b][c];vis[k][a][b][c]=1;
15     f[k][a][b][c]=0;
16     int s=a+b+c+1;
17     if(a+b+c<7) f[k][a][b][c]+=1.0*c/s*ffind(k-1,a,b+1,c),
18                 f[k][a][b][c]+=1.0*b/s*ffind(k-1,a+1,b-1,c+1);
19     else f[k][a][b][c]=1.0*c/s*ffind(k-1,a,b+1,c-1),f[k][a][b][c]+=1.0*b/s*ffind(k-1,a+1,b-1,c);
20     f[k][a][b][c]+=1.0*a/s*ffind(k-1,a-1,b,c);
21     f[k][a][b][c]+=1.0/s*(ffind(k-1,a,b,c)+1);
22     return f[k][a][b][c];
23 }
24 
25 int main()
26 {
27     int T;
28     scanf("%d",&T);
29     while(T--)
30     {
31         int k,a,b,c;
32         scanf("%d%d%d%d",&k,&a,&b,&c);
33         // memset(vis,0,sizeof(vis));
34         for(int i=0;i<=k;i++) for(int j=0;j<=7;j++) for(int jj=0;jj<=7-j;jj++) for(int jjj=0;jjj<=7-j-jj;jjj++) vis[i][j][jj][jjj]=0;
35         printf("%.2lf\n",ffind(k,a,b,c));
36     }
37     return 0;
38 }