python: add problem 59 and unittest

Author: supercoderhawk

Python/sln_1_100/solution_51_60.py

@@ -2,6 +2,7 @@
 from itertools import permutations
 from common_utils import Interval
 
+
 class Solution_51_60:
     def solveNQueens(self, n):
         """
@@ -60,6 +61,7 @@ def spiralOrder(self, matrix):
             curr_direction = (curr_direction + 1) % 4
 
         return res
+
     def merge(self, intervals):
         """
         56
@@ -80,6 +82,7 @@ def merge(self, intervals):
             new_intervals.append(Interval(start, end))
             last_start, last_end = start, end
         return new_intervals
+
     def lengthOfLastWord(self, s):
         """
         58
@@ -106,3 +109,29 @@ def lengthOfLastWord(self, s):
             last_ch = cur_ch
             index -= 1
         return end - start
+
+    def generateMatrix(self, n):
+        """
+        59
+        :param n:
+        :return:
+        """
+        x, y = 0, -1
+        directions = [[0, 1], [1, 0], [0, -1], [-1, 0]]
+        steps = [n, n - 1]
+
+        val = 1
+        matrix = [[0] * n for _ in range(n)]
+        direction_index = 0
+
+        while steps[direction_index % 2]:
+            curr_direction = directions[direction_index]
+            for _ in range(steps[direction_index % 2]):
+                x += curr_direction[0]
+                y += curr_direction[1]
+                matrix[x][y] = val
+                val += 1
+            steps[direction_index % 2] -= 1
+            direction_index = (direction_index + 1) % 4
+
+        return matrix

Python/sln_1_100/test_solution_51_60.py

@@ -2,6 +2,7 @@
 from sln_1_100.solution_51_60 import Solution_51_60
 from common_utils import Interval
 
+
 # -*- coding: UTF-8 -*-
 class TestSolution_51_60(TestCase):
     def setUp(self):
@@ -12,12 +13,17 @@ def test_spiralOrder(self):
         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_merge(self):
-        raw_span1 = [[1,3],[2,6],[8,10],[15,18]]
-        span1 = [Interval(start,end) for start,end in raw_span1]
+        raw_span1 = [[1, 3], [2, 6], [8, 10], [15, 18]]
+        span1 = [Interval(start, end) for start, end in raw_span1]
 
     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)
+
+    def test_generateMatrix(self):
+        matrix = self.sln.generateMatrix(3)
+        self.assertEqual(matrix, [[1, 2, 3], [8, 9, 4], [7, 6, 5]])