Dice HDU - 4652(期望）

题目地址：http://acm.hdu.edu.cn/showproblem.php?pid=4652
题意：有一个m面的骰子，分别求出现最后n次点数相同与最后n次点数各不同的期望次数。
思路：
期望DP
f[i]=∑p[i - j] * (f[j]+w[i-j])
p[i-j]表示状态i到状态j的概率，f[i]表示状态i到达终点状态的期望，w[i-j]表示状态i转移到状态j所贡献的期望值

• 最后n次点数相同
  f[0]=f[1]+1
  f[i]=1/m* f[i+1]+(m-1)/m * f[1]+1 ①
  f[i+1]=1/m* f[i+2]+(m-1)/m* f[1]+1 ②
  ①-②得f[i]-f[i+1]=1/m*(f[i+1]-f[i+2])
  f[i+1]-f[i+2]=m*(f[i]-f[i+1])
  由此发现 f[i+1]-f[i]是首项为1，等比为m的等比数列
  而f[1]-f[0]=1;
  那么f[2]-f[1]=m f[i+1]-f[i]=mⁱ
  并且f[n]=0
  那么f[0]=1+m+…+m^n-1
• 最后n次点数各不相同

#include <cstdio>
#include <algorithm>
#include <iostream>
#include <cstring>
using namespace std;

int main()
{
   int t;
   while(scanf("%d",&t)!=EOF)
   {
       while(t--)
       {
           int op,m,n;∑
           double ans;
           ans=0;
           scanf("%d%d%d",&op,&m,&n);
           if(op==0)
           {
               double x=1.0;
               for(int i=0;i<n;i++)
               {
                   ans+=x;
                   x=x*m;
               }
           }
           else
           {
               double x=1.0;
               for(int i=1;i<=n;i++)
               {
                   ans+=x;
                   x=x*m/(double)(m-i);
               }
           }
           printf("%.9lf\n",ans);
       }
   }
    return 0;
}