hdu-4586数学期望

题目链接：https://cn.vjudge.net/problem/HDU-4586
参考：https://blog.csdn.net/tianyuhang123/article/details/77937335
题意：给你一个骰子，每个面上有个数字代表可以赢得钱，接下来有几个带着特殊颜色的面，如果此面朝上，则可以获得继续扔一次的机会并获得钱，求获得钱的数学期望。
思路：设骰子钱数总和为sum，n为总面数，m为特殊颜色的面数，则扔一次获得的钱为sum/n，扔两次的钱为sum/n*（m/n）；扔三次：sum/n*（m/n）²……，可以看出是一个等比数列。此时需要分三种情况，一定要注意先后讨论的顺序。1.sum=0，则钱数为0 2.m=n，则等比数列的q=1，和为无穷大 3.m<n 则求和时(m/n)^k是趋向于0的，因此答案为sum/(n-m）。

#include <bits/stdc++.h>
using namespace std;
int main()
{
    int n,m;
    while(~scanf("%d",&n))
    {
        double ans=0;
        int x;
        for(int i=0; i<n; i++)
        {
            scanf("%d",&x);
            ans+=x;
        }
        scanf("%d",&m);
        for(int i=0; i<m; i++)
        {
            scanf("%d",&x);
        }
        if(ans==0)
            printf("0.00\n");
        else if(n==m)
            printf("inf\n");
        else
            printf("%.2f\n",ans/(n-m));
    }
    return 0;
}