solve Length of Last Word

Author: saubhik

problems/length_of_last_word.py

@@ -0,0 +1,15 @@
+class Solution:
+    # Time: O(n)
+    # Space: O(1)
+    def lengthOfLastWord(self, s: str) -> int:
+        curr_len = last_len = 0
+        for char in s:
+            if char == " " and curr_len > 0:
+                last_len, curr_len = curr_len, 0
+            elif char != " ":
+                curr_len += 1
+
+        if curr_len > 0:
+            last_len = curr_len
+
+        return last_len

tests/test_length_of_last_word.py

@@ -0,0 +1,14 @@
+import unittest
+
+from length_of_last_word import Solution
+
+
+class TestLengthOfLastWord(unittest.TestCase):
+    def test_example_1(self):
+        assert Solution().lengthOfLastWord(s="Hello World") == 5
+
+    def test_example_2(self):
+        assert Solution().lengthOfLastWord(s=" Hello   World    ") == 5
+
+    def test_example_3(self):
+        assert Solution().lengthOfLastWord(s="   ") == 0