#include <iostream>


void PrintType(int type)
{
    using namespace std;

    switch (type)
    {
        case 1:
            cout << "Total Population";
            break;
        case 2:
            cout << "Batman";
            break;
        case 3:
            cout << "Joker";
            break;
        case 4:
            cout << "Faction Mafia";
            break;
        case 5:
            cout << "Mafia Doctor";
            break;
        case 6:
            cout << "Patrolmen";
            break;
        case 7:
            cout << "Boondock";
            break;
        case 8:
            cout << "Pardoner";
            break;
        case 9:
            cout << "Doctor";
            break;
        case 10:
            cout << "Director";
            break;
        case 11:
            cout << "Regular Townie";
            break;
        case 12:
            cout << "Faction Townie";
            break;
        case 13:
            cout << "Special Townie";
            break;
        default:
            cout << "whoopsies";
    }
}




int main()
{
    using namespace std;

    double player_count[20];
    double probability[20];
    for(int i = 0; i <=19; i++)
    {
        player_count[i] = 1337;
        probability[i] = 1337;
    }
    /*
    index 1 = total pop
    index 2 = batman
    index 3 = joker
    index 4 = faction mafia
    index 5 = mafia doctor
    index 6 = patrolmen
    index 7 = boondock
    index 8 = pardoner
    index 9 = doctor
    index 10 = director
    index 11 = regular townie
    index 12 = faction townie
    index 13 = special townie
    */

   

    cout << "Input Numbers of Players" << endl;
    for(int i = 1; i <= 10; i++)
    {
        cout << endl;
        PrintType(i);
        cout << ": ";
        cin >> player_count[i];
    }

    player_count[12] = player_count[1] - player_count[4] - player_count[3];                                                        //faction townie = total pop - mafia faction - joker
    player_count[13] = player_count[10] + player_count[9] + player_count[8] + player_count[7] + player_count[6] + player_count[2]; //special townie is sum of special roles
    player_count[11] = player_count[12] - player_count[13];                                                                        //regular townie = faction townie - special townie

   


    /*
    probability that a given group will die is based on table

    ------group------other
    group---W----------X
    other---Y----------Z

    group/group + 2*other/group = probability at least 1 of that group will get hit
    W + 2*X = W + X + Y = W + 2*Y
    we exclude other/other since group is not part of that

    */
   

    double W;
    double X;

    for(int i = 4; i <= 13; i++)
    {
        W = (player_count[i]/(player_count[1] - player_count[2] - player_count[3]))*           //group/population - bat&joker
            ((player_count[i]-1)/(player_count[1] - player_count[2] - player_count[3] - 1));   //group lost a member/population lost a member

        X = (player_count[i]/(player_count[1] - player_count[2] - player_count[3]))*                                                                  //group/population
            ((player_count[1] - player_count[i] - 1 - player_count[2] - player_count[3])/(player_count[1] - player_count[2] - player_count[3] - 1));  //other - dead player - joker&bat / total-joker&bat
       
        probability[i] = W + 2*X;

        cout << endl
            << "Probability of at least 1 ";
        PrintType(i);
        cout << " dying is: "
            << 100*probability[i];
    }

    cout << endl
        << "Probability of any given individual being mafia: "
        << 100 * player_count[4] / player_count[1];

    cout << endl;
    return 0;
}