概率dp求期望-hdu-4586-Play the Dice

题目链接：

http://acm.hdu.edu.cn/showproblem.php?pid=4586

题目大意：

有一个色子，n面，每面有个分值a[i],其中有m面比较特殊，当该面出现时，可以再投一次。求最后得分期望。

解题思路：

设所求期望为ans.

考虑投掷一次的情况，如果是m当中的某一面i时，相当于期望增加了a[i],其它所有条件没变，相当于这次没投掷。当投掷到其他面时，得到相应分值。

所以ans=1/n*(a[b[1]]+ans)+1/n*(a[b[2]]+ans)+..1/n*(a[b[m]]+ans)+1/n*a[k]

化简得ans=sum/(n-m) 容易知道m面是等价的。

当sum为0是，为0.sum不为0，而n==m时，ans=INF

代码：

#include<iostream>#include<cmath>#include<cstdio>#include<cstdlib>#include<string>#include<cstring>#include<algorithm>#include<vector>#include<map>#include<set>#include<stack>#include<list>#include<queue>#define eps 1e-6#define INF 0x1f1f1f1f#define PI acos(-1.0)#define ll __int64#define lson l,m,(rt<<1)#define rson m+1,r,(rt<<1)|1#pragma comment(linker, "/STACK:1024000000,1024000000")using namespace std;//freopen("data.in","r",stdin);//freopen("data.out","w",stdout);int main(){   int n,m;   while(~scanf("%d",&n))   {      int sum=0,a;      for(int i=1;i<=n;i++)         scanf("%d",&a),sum+=a;      scanf("%d",&m);      for(int i=1;i<=m;i++)         scanf("%d",&a);      if(sum==0)         printf("0.00\n");      else if(n==m)         printf("inf\n");      else         printf("%.2f\n",sum*1.0/(n-m));   }   return 0;}