Update Arith Geo

Author: abx20

Arith Geo

@@ -35,32 +35,24 @@
 *                                                                                      *
 ***************************************************************************************/
 
-function ArithGeo(arr) { 
+function ArithGeo(arr) {
+  if (arr.length < 2) return -1;
 
-  var arithFlag = true, geoFlag = true;
-  var diff = arr[1] - arr[0];
-  
-  for (var i = 2; i < arr.length; i++) {
-    if ((arr[i] - arr[i-1]) !== diff) {
-      arithFlag = false;
-    }
-  }
-  if (arithFlag) {
-    return "Arithmetic";
-  }
-  else { // check for geometric pattern
-    diff = arr[1] / arr[0];
-    for (var i = 2; i < arr.length; i++) {
-      if ((arr[i] / arr[i-1]) !== diff) { 
-        geoFlag = false;
-      }
-    }
-    if (geoFlag) {
-      return "Geometric";
+  let isArithmetic = true;
+  let isGeometric = true;
+  let diff = arr[1] - arr[0];
+  let ratio = arr[1] / arr[0];
+
+  for (let i = 1; i < arr.length; i++) {
+    if (arr[i] - arr[i - 1] !== diff) {
+      isArithmetic = false;
     }
-    else {
-      return "-1";
+    if (arr[i] / arr[i - 1] !== ratio) {
+      isGeometric = false;
     }
   }
-  
+
+  if (isArithmetic) return "Arithmetic";
+  if (isGeometric) return "Geometric";
+  return -1;
 }