【JZOJ100004】【NOI2017模拟.4.1】 Dice

任务

解法

我们分开考虑每个骰子的贡献；
设$X_i$表示第$i$个骰子的点数。
显然$Ans1=\sum_{i=1}^nE[X_i]$，
又$E[X_i]=\sum_{j=1}^6j*P[X_i=j]$，
但题目要求我们不能连续两次投相同的骰子；
所以$P[X_i=j]=\sum_{k=1}^6P[X_{i-1}=k]*P[X_i=j|P_{i-1}=k]$，其中$i>1$

$$P[X_i=j|P_{i-1}=k]=\begin{cases} 0,&j=k\\ p_j+p_k*P[X_i=j|P_{i-1}=k],&j！=k \end{cases}$$
所以 $P[X_i=j|P_{i-1}=k]=\frac{p_j}{1-p_k}$

---

我们知道$E[Ans1]$后，要求$Ans2=E[(Ans1-E[Ans1])^2]$；
也即$Ans2=E[(\sum X_i-E[Ans1])^2]$
由于$E[Ans1]$是常量，我们先让$X_i-\frac{E[ans1]}{n}$，则有$Ans2=E[(\sum X'_i)^2]$
所以

$$\begin{align}E[Ans2]&=E[(\sum X'_i)^2]\\ &=E[\sum_{i=1}^n\sum_{j=1}^nX_i*X_j]\\ &=E[\sum_{i=1}^n(X_i)^2]+2*E[\sum_{i=1}^n\sum_{j=i+1}^nX_i*X_j] \end{align}$$
容易算出 $E[\sum_{i=1}^n(X_i)^2]$，
我们考虑 $E[\sum_{i=1}^n\sum_{j=i+1}^nX_i*X_j]$；
$$\begin{align} E[\sum_{i=1}^n\sum_{j=i+1}^nX_i*X_j]&=\sum_{i=1}^n\sum_{j=i+1}^n\sum_{k=1}^6\sum_{l=1}^6k*l*P[X_i=k]*P[X_j=l|X_i=k]\\ &=\sum_{k=1}^6\sum_{l=1}^6k*l*\sum_{i=1}^n\sum_{j=i+1}^nP[X_i=k]*P[X_j=l|X_i=k]\\ &=\sum_{k=1}^6\sum_{l=1}^6k*l*\sum_{i=1}^nP[X_i=k]*\sum_{j=i+1}^nP[X_j=l|X_i=k]\\ &=\sum_{k=1}^6\sum_{l=1}^6k*l*\sum_{i=1}^nP[X_i=k]*\sum_{j=1}^{n-i+1}P[X_j=l|X_1=k] \end{align}$$
由于 $\sum_{j=1}^{n-i+1}P[X_j=l|X_1=k]$只与三元有关，所以可以预处理。

代码

#include<iostream>
#include<algorithm>
#include<math.h>
#include<stdio.h>
#include<string.h>
#define ll long long
#define db double
using namespace std;
const char* fin="dice.in";
const char* fout="dice.out";
const int inf=0x7fffffff;
const int maxn=100007;
int n,m,i,j,k,l;
db p[7],ans1=0,ans2=0,oo;
db f[maxn][7],g[7][7][maxn],h[7][7][maxn];
int main(){
    freopen(fin,"r",stdin);
    freopen(fout,"w",stdout);
    for (i=1;i<=6;i++) scanf("%lf",&p[i]);
    scanf("%d",&n);
    for (i=1;i<=n;i++)
        for (j=1;j<=6;j++)
            if (i==1) f[i][j]=p[j];
            else for (k=1;k<=6;k++) 
                if (j!=k) f[i][j]+=f[i-1][k]*(p[j]/(1-p[k]));
    for (i=1;i<=n;i++) for (j=1;j<=6;j++) ans1+=f[i][j]*j;
    printf("%lf\n",ans1);
    for (i=1;i<=6;i++) for (j=1;j<=6;j++){
        if (i!=j){
            g[i][j][2]=p[j]/(1-p[i]);
            h[i][j][2]=g[i][j][2];
        }
    }
    for (k=3;k<=n;k++){
        for (i=1;i<=6;i++) for (j=1;j<=6;j++){
            for (l=1;l<=6;l++){
                if (j!=l){
                    g[i][j][k]+=g[i][l][k-1]*(p[j]/(1-p[l]));
                }
                h[i][j][k]=g[i][j][k]+h[i][j][k-1];
            }
        }
    }
    db C=ans1/n;
    for (k=1;k<=6;k++)
        for (l=1;l<=6;l++)
            for (i=1;i<n;i++)
                ans2+=(k-ans1/n)*(l-ans1/n)*f[i][k]*h[k][l][n-i+1];
    ans2*=2;
    for (i=1;i<=n;i++) for (j=1;j<=6;j++) ans2+=(j-ans1/n)*(j-ans1/n)*f[i][j];
    printf("%lf\n",ans2);
    return 0;
}