1 files changed, 15 insertions, 16 deletions
diff --git a/ee1301/wk3/hw3_directory/strap012_HW3A.cpp b/ee1301/wk3/hw3_directory/strap012_HW3A.cpp
index fd051fd..e22ef89 100644
--- a/ee1301/wk3/hw3_directory/strap012_HW3A.cpp
+++ b/ee1301/wk3/hw3_directory/strap012_HW3A.cpp
@@ -11,33 +11,32 @@ One-armed Bandit Simulator
 #include <iostream>
 #include <stdlib.h>
 #include <cmath>
-#include <iomanip> 
+#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--) {
+  for (int i=1; i<w; i++) {
     spinNew = rand() % d + 1;
-    if (spinNew==spinOG) {
-      //Do nothing
-    } else {
+    if (spinNew!=spinOG) {
       return 0;
     }
   }
   return 1;
 }
 int main () {
-    long n=0;
-    int m=0;
-    float d= 3;
-    float w= 9;
-    for (n; n<=1000000; n++) {
-      m = spin_the_wheel(d,w);
-      if (m==1) {
-        m++;
-      }
+  long n = 0;
+  int m = 0;
+  int test;
+  int w = 3;
+  int d = 9;
+  for (n; n <= 1000; n++)
+  {
+    test = spin_the_wheel(d, w);
+    m+=test;
     }
     cout << "w=" << w << ", d=" << d;
-    cout << ": Simulated probability = m/n = " << (m/n)*100 << ". ";
-    cout << "Theoretical probability = " << (d/pow(d,w))*100 << "." << endl;
+    cout << ": Simulated probability = m/n = " << (m/n)*100 << "%. ";
+    cout << "Theoretical probability = " << (d/(pow(d,w)))*100 << "%." << endl;
 }
 \ No newline at end of file