Update Mojo code

Author: uberkael

bench.sh

@@ -9,7 +9,7 @@ for dir in *; do
 		cd "$dir"
 		echo "Benchmarking $dir"
 		$COMPILE
-		$BENCH  "${COMMANDS[@]}"
+		$BENCH "${COMMANDS[@]}"
 		cd ..
 	fi
 done

problem-1/problem-1.mojo

@@ -1,22 +1,14 @@
-from python import Python
-
-def lengthOfLastWord(enterword: String):
-    
-    py = Python.import_module("builtins")
-    
-    count = 0
+fn lengthOfLastWord(enterword: String):
+    var count: Int = 0
     for i in range(len(enterword)-1, -1, -1):
         if enterword[i] != " ":
             count += 1
         elif count > 0:
             break
-    
-    print("Length of the last word:")
-    print(count)
-
 
-def main():
-    input_string = "Hello World" 
-    lengthOfLastWord(input_string)
+    print("Length of the last word:", count)
 
 
+fn main():
+    var input_string = "Hello World"
+    lengthOfLastWord(input_string)

problem-2/problem-2.mojo

@@ -1,14 +1,11 @@
-from python import Python
+fn checkRecord(attendance_record: String) -> Bool:
+    var attendance_record_str = attendance_record
 
-def checkRecord(attendance_record: String):
-    py = Python.import_module("builtins")
-    attendance_record_str = attendance_record
-
-    absent_count = 0
-    late_count = 0
+    var absent_count = 0
+    var late_count = 0
 
     for i in range(len(attendance_record_str)):
-        
+
         if attendance_record_str[i] == 'A':
             absent_count += 1
             late_count = 0
@@ -25,8 +22,5 @@ def checkRecord(attendance_record: String):
     return True
 
 
-def main():
-    checkRecord("PAALP")
-
-
-
+fn main():
+    var value = checkRecord("PAALP")

problem-3/problem-3.mojo

@@ -1,25 +1,23 @@
-from python import Python
+import math
 
-def makeGood(s: String):
-    py = Python.import_module("builtins")
-    result = str("")
+fn makeGood(s: String):
+    var result = str("")
 
-    i = 0
+    var i = 0
     while i < len(s):
-        
-        if i < len(s) - 1 and py.abs(ord(s[i]) - ord(s[i + 1])) == 32:
-            
+
+        if i < len(s) - 1 and math.abs(ord(s[i]) - ord(s[i + 1])) == 32:
+
             i += 2
         else:
-           
+
             result = result + s[i]
             i += 1
 
     print(result)
 
-def main():
-    
-    input_str = "leEeetcode"
+fn main():
+    var input_str = "leEeetcode"
     makeGood(input_str)
-    
+
     #print("After making the string good:", result)

problem-4/problem-4.mojo

@@ -1,36 +1,20 @@
-from python import Python
+fn detectCapitalUse(word: String) -> Bool:
+    var uinput = word
 
-def detectCapitalUse(word: String):
-    py = Python.import_module("builtins")
-    
-    uinput = word
-
-    number_of_capital_letters = 0
+    var number_of_capital_letters = 0
     for i in range(len(uinput)):
-        letter = uinput[i]
+        var letter = uinput[i]
         if ord(letter) < 97:
             number_of_capital_letters += 1
 
-    number_of_small_letters = len(uinput) - number_of_capital_letters
+    var number_of_small_letters = len(uinput) - number_of_capital_letters
 
     if number_of_capital_letters == len(uinput) or number_of_small_letters == len(uinput) or (ord(word[0]) < 97 and number_of_capital_letters == 1):
         return True
     else:
         return False
 
-def main():
-    result = detectCapitalUse("USA")
+fn main():
+    var result = detectCapitalUse("USA")
     print("Result:")
     print(result)
-
-
-
-
-
-
-
-
-
-
-
-

problem-5/problem-5.mojo

@@ -1,14 +1,10 @@
-from python import Python
-
-def findTheDifference(s:String, t:String):
-    py = Python.import_module("builtins")
-    
+fn findTheDifference(s:String, t:String):
     for i in range(len(t)):
         if s[i] != t[i]:
-            result = t[i]
+            var result = t[i]
             print("The difference is:")
             print(result)
             break
 
-def main():
+fn main():
     findTheDifference("abcd", "abced")