Cod sursa(job #2504663)

Utilizator adimiclaus15Miclaus Adrian Stefan adimiclaus15 Data 5 decembrie 2019 12:44:56
Problema Dezastru Scor 30
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.94 kb
#include <bits/stdc++.h>
#define NMAX 26
using namespace std;
ifstream f("dezastru.in");
ofstream g("dezastru.out");
int n,k;
long long comb[NMAX][NMAX];
double prob[NMAX],sol;
int nr_biti(int x)
{
    int nr=0;
    do
    {
        nr++;
        x=x&(x-1);
    }
    while(x);
    return nr;
}
int main()
{
    f>>n>>k;
    for(int i=0;i<n;i++)
        f>>prob[i];
    for(int i=1;i<=n;i++)
    {
        comb[i][1]=i;
        comb[i][i]=1;
    }
    for(int i=2;i<=n;i++)
    {
        for(int j=2;j<i;j++)
            comb[i][j]=comb[i-1][j]+comb[i-1][j-1];
    }
    for(int mask=1;mask<(1<<n);mask++)
    {
        if(nr_biti(mask)==k)
        {
            double prod=1;
            for(int i=0;i<n;i++)
            {
                if(mask&(1<<i))
                    prod*=prob[i];
            }
            sol+=prod/comb[n][k];
        }
    }
    g<<fixed<<setprecision(7)<<sol;
    return 0;
}