ee1301/wk3/hw3_directory/strap012_HW3A.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

/*
27 Feb 2019
Matthew Strapp
5449340
EE1301
Spring 2019
Homework 3A
One-armed Bandit Simulator
*/

#include <iostream>
#include <stdlib.h>
#include <cmath>
#include <iomanip> 
using namespace std;
int spin_the_wheel(int d, int w) {
  int spinOG, spinNew;
  spinOG = rand() % d + 1;
  for (int i=w; i>0; i--) {
    spinNew = rand() % d + 1;
    if (spinNew==spinOG) {
      //Do nothing
    } else {
      return 0;
    }
  }
  return 1;
}
int main () {
    long n=1000000;
    int m=0;
    int d= 3;
    int w= 9;
    for (n; n>0; n--) {
      m = spin_the_wheel(d,w);
      if (m==1) {
        m++;
      }
    }
    cout << "w=" << w << ", d=" << d;
    cout << ": Simulated probability = m/n = " << (m/n)*100 << ". ";
    cout << "Theoretical probability = " << (d/pow(d,w))*100 << "." << endl;
}