Lightoj 1027 - A Dangerous Maze

题意：初始地有n扇门，每次等概率选择一扇门，每扇门有会在$x_i$时间内出去，或者回到原点。$x_i$为正数时$x_i$时间后可以出去，负数时$x_i$后回到初始地。求出去需要时间的期望。

思路：每次选择门要么可以出去，要么回到原点，假设$positive$代表$x_i$为正数的数量，$sum$代表$x_i$的和，那么出去的概率就为$\frac{positive}{n}$，回到原点的概率为$1-\frac{positive}{n}$，每次选择门平均花费的时间就为$\frac{sum}{n}$。显然满足几何分布，根据几何分布期望公式出去的期望就为$（\frac{sum}{n}）\ast （\frac{n}{positive}）。$

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
int a[105];
int main()
{
    int t,n,s=0;
    cin>>t;
    while(t--)
    {
        cin>>n;
        int positive=0,sum=0;
        for(int i=1;i<=n;i++)
        {
            cin>>a[i];
            sum+=abs(a[i]);
            if(a[i]>0) positive++;
        }
        cout<<"Case "<<++s<<": ";
        if(!positive) cout<<"inf"<<endl;
        else
        {
            cout<<sum/__gcd(sum,positive)<<"/"<<
            positive/__gcd(sum,positive)<<endl;
        }
    }
}