python: add problem 58 and unittest

Author: supercoderhawk

Python/solution_51_60.py renamed to Python/sln_1_100/solution_51_60.py

@@ -41,3 +41,30 @@ def spiralOrder(self, matrix):
             curr_direction = (curr_direction + 1) % 4
 
         return res
+
+    def lengthOfLastWord(self, s):
+        """
+        58
+        :type s: str
+        :rtype: int
+        """
+        if len(s) <= 1:
+            if not s or s == ' ':
+                return 0
+            else:
+                return 1
+        start = end = len(s)
+        index = len(s) - 2
+        last_ch = s[-1]
+        while index >= 0:
+            cur_ch = s[index]
+            if last_ch == ' ' and cur_ch != ' ':
+                end = index + 1
+            elif cur_ch == ' ' and last_ch != ' ':
+                start = index + 1
+                break
+            if index == 0 and cur_ch != ' ':
+                start = index
+            last_ch = cur_ch
+            index -= 1
+        return end - start

Python/sln_1_100/test_solution_51_60.py

@@ -0,0 +1,20 @@
+from unittest import TestCase
+from sln_1_100.solution_51_60 import Solution_51_60
+
+
+# -*- coding: UTF-8 -*-
+class TestSolution_51_60(TestCase):
+    def setUp(self):
+        self.sln = Solution_51_60()
+
+    def test_spiralOrder(self):
+        m = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
+        res = [1, 2, 3, 6, 9, 8, 7, 4, 5]
+        self.assertEqual(self.sln.spiralOrder(m), res)
+        self.assertEqual(self.sln.spiralOrder([[1]]), [1])
+
+    def test_lengthOfLastWord(self):
+        length = self.sln.lengthOfLastWord('Hello world')
+        self.assertEqual(length, 5)
+        self.assertEqual(self.sln.lengthOfLastWord('  '), 0)
+        self.assertEqual(self.sln.lengthOfLastWord('a  '), 1)