Edit my solution

Author: asubramanian08

Week30.ipynb

@@ -27,7 +27,7 @@
     "\n",
     "Think about what the problem is asking, break it down\n",
     "\n",
-    "The first step is determining which dice wins.\n",
+    "First we must code up the `beats()` function.\n",
     "1. Count the number of ways dice A wins.\n",
     "2. Count the number of ways dice B wins.\n",
     "3. Whoever have a larger value is the winner. If both win the same number of times, that dice are tied."
@@ -39,7 +39,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "\"\"\" Python solution to USACO January 2022 Bronze problem two\n",
+    "\"\"\" Python solution to USACO January 2022 Bronze problem two.\n",
     "Translated into python from my c++ solution. Below is the link to my original code.\n",
     "https://github.com/asubramanian08/USACO/blob/master/Bronze/2021-2022/JAN2022/q2.cpp \"\"\"\n",
     "\n",
@@ -57,23 +57,28 @@
     "            bWins += A[i] < B[j]\n",
     "    return aWins - bWins\n",
     "\n",
+    "# Get input\n",
     "C = [0, 0, 0, 0]  # Using 0 as place holder\n",
     "for test in range(int(input())):\n",
     "    diceFaces = list(map(int, input().split(' ')))\n",
     "    A = diceFaces[:SIDES]\n",
     "    B = diceFaces[SIDES:]\n",
-    "    foundAnswer = False\n",
+    "    \n",
+    "    # Unify cycle direction\n",
     "    winner = beats(A, B)\n",
-    "    if winner == 0:  # Tie\n",
-    "        foundAnswer = True\n",
-    "    elif winner < 0:  # B won\n",
+    "    if winner < 0:\n",
     "        A, B = B, A\n",
+    "    \n",
+    "    # Determine is there is a valid cycle\n",
+    "    foundAnswer = False\n",
     "    for C[0] in range(MIN_FACE, MAX_FACE + 1, 1):\n",
     "        for C[1] in range(MIN_FACE, MAX_FACE + 1, 1):\n",
     "            for C[2] in range(MIN_FACE, MAX_FACE + 1, 1):\n",
     "                for C[3] in range(MIN_FACE, MAX_FACE + 1, 1):\n",
     "                    if beats(C, A) > 0 and beats(B, C) > 0:\n",
     "                        foundAnswer = True  # Cycle found\n",
+    "    \n",
+    "    # Print yes when there is no tie and a cycle has been found\n",
     "    print(\"yes\" if (winner != 0 and foundAnswer) else \"no\")"
    ]
   }