#include "stdint.h"
#include "TMath.h"
#include <cmath>
#include <TRandom1.h>
#include <iostream>

//Varied Monty Hall Program!

//Function to return a random number to serve as a pointer in the 'door_array' array
int RandomDoor(int& doors)
{
    TRandom1 *myran=new TRandom1(0,389);
    int returnVal=myran->Rndm()*doors;
    return returnVal;
}

//Function to Search for an available door
int Search(int* array,int& doors,int& choosedoor,int& host_doors)
{
    int returnVal;
    if ((doors-host_doors)==1) {
        returnVal=choosedoor;
    }
    
    int rNum=RandomDoor(doors);
    if ((array[rNum]==2)||(rNum==choosedoor)) {
        rNum=RandomDoor(doors);
        while ((array[rNum]==2)||(rNum==choosedoor)) {
            if ((rNum==(doors-1))&&(array[rNum]==2)) {
                rNum=0;
            }
            if ((rNum==(doors-1))&&(rNum==choosedoor)) {
                rNum=0;
            }
            else
            {
                rNum++;
            }
        }
        returnVal=rNum;
    }
    returnVal=RandomDoor(doors);
    return returnVal;
}

void MontyHall2(int doors, long iterations, int host_doors)
{
    double host_wins=0;
    double wins=0;
    double winpct=0;
    //generate array
    int* door_array=new int[doors];
    
    //loop through game 'iteratons' times
    for (long i=0; i<iterations; i++) {
        
        //populate array with 0's
        
        for (int j=0; j<doors; j++) {
            door_array[j]=0;
        }
        
        //choose winning door
        door_array[RandomDoor(doors)]=1;
        
        //player randomly chooses
        
        int choosedoor=RandomDoor(doors);
        
        //Host randomly chooses doors and turns them to "2"
        
        bool rightVal=false;
        for (int k=0; k<host_doors; k++) {
            int temp=Search(door_array,doors,choosedoor,host_doors);
            door_array[temp]=2;
            if (door_array[temp]==1) {
                rightVal=true;
            }
        }
        
        //If host chooses correctly, host_wins++ and game over
        if (rightVal==true) {
            host_wins++;
        }
        
        else {
            
            //Player switches doors
            int limit;
            if (host_doors>3) {
                limit=host_doors/3;
            }
            else {
                limit=1;
            }
            
            bool playerwins=false;
            for (int n=0; n<limit; n++) {
                if (door_array[Search(door_array,doors,choosedoor,host_doors)]==1) {
                    playerwins=true;
                }
            }
            if (playerwins==true) {
                wins++;
            }
        }
    }
    winpct=(100*wins)/iterations;
    cout<<winpct<<" %"<<endl;
    delete[] door_array;
}