def test_subdomainVisits(self):
solu = ArraySolution()
A = ['9001 discuss.leetcode.com']
res = solu.subdomainVisits(A)
self.assertEqual(len(res), 3)
for str in res:
print(str)
def test_minimumAbsDifference(self):
sole = ArraySolution()
self.assertEqual([[1, 2], [2, 3], [3, 4]],
sole.minimumAbsDifference([4, 2, 1, 3]))
self.assertEqual([[1, 3]], sole.minimumAbsDifference([1, 3, 6, 10,
15]))
self.assertEqual([[-14, -10], [19, 23], [23, 27]],
sole.minimumAbsDifference(
[3, 8, -10, 23, 19, -4, -14, 27]))
def test_spiralMatrixIII(self):
sole = ArraySolution()
self.assertEqual([[0, 0], [0, 1], [0, 2], [0, 3]],
sole.spiralMatrixIII(1, 4, 0, 0))
self.assertEqual(
[[1, 4], [1, 5], [2, 5], [2, 4], [2, 3], [1, 3], [0, 3], [0, 4],
[0, 5], [3, 5], [3, 4], [3, 3], [3, 2], [2, 2], [1, 2], [0, 2],
[4, 5], [4, 4], [4, 3], [4, 2], [4, 1], [3, 1], [2, 1], [1, 1],
[0, 1], [4, 0], [3, 0], [2, 0], [1, 0], [0, 0]],
sole.spiralMatrixIII(5, 6, 1, 4))
def test_surfaceArea(self):
sole = ArraySolution()
self.assertEqual(10, sole.surfaceArea([[2]]))
self.assertEqual(34, sole.surfaceArea([[1, 2], [3, 4]]))
self.assertEqual(16, sole.surfaceArea([[1, 0], [0, 2]]))
self.assertEqual(32, sole.surfaceArea([[1, 1, 1], [1, 0, 1], [1, 1,
1]]))
self.assertEqual(46, sole.surfaceArea([[2, 2, 2], [2, 1, 2], [2, 2,
2]]))
def test_sortArrayByParity(self):
A = [4, 2, 5, 7]
solu = ArraySolution()
p = solu.sortArrayByParityII(A)
for i in p:
print(i)
def test_canVisitAllRooms(self):
sole = ArraySolution()
self.assertTrue(sole.canVisitAllRooms([[1], [2], [3], []]))
def test_escapeGhosts(self):
sole = ArraySolution()
self.assertFalse(
sole.escapeGhosts([[1, 8], [-9, 0], [-7, -6], [4, 3], [1, 3]],
[6, -9]))
def test_maxSumAfterPartitioning(self):
sole = ArraySolution()
self.assertEqual(
84, sole.maxSumAfterPartitioning([1, 15, 7, 9, 2, 5, 10], 3))
def test_kClosest(self):
sole = ArraySolution()
print(sole.kClosest([[3, 3], [5, -1], [-2, 4]], 2))
def test_pathInZigZagTree(self):
sole = ArraySolution()
self.assertEqual([1, 3, 4, 14], sole.pathInZigZagTree(14))
self.assertEqual([1, 2, 6, 10, 26], sole.pathInZigZagTree(26))
def test_minCostToMoveChips(self):
sole = ArraySolution()
self.assertEqual(1, sole.minCostToMoveChips([1, 2, 3]))
self.assertEqual(2, sole.minCostToMoveChips([2, 2, 2, 3, 3]))
def test_subarraySum(self):
sole = ArraySolution()
self.assertEqual(2, sole.subarraySum([1, 1, 1], 2))
def test_findDuplicate(self):
sole = ArraySolution()
self.assertEqual(2, sole.findDuplicate([1, 3, 4, 2, 2]))
self.assertEqual(3, sole.findDuplicate([1, 3, 4, 3, 2]))
self.assertEqual(4, sole.findDuplicate([1, 3, 4, 2, 4]))
def test_totalNQueens(self):
sole = ArraySolution()
self.assertEqual(2, sole.totalNQueens(4))
def test_duplicateZeros(self):
sole = ArraySolution()
arr = [1, 0, 2, 3, 0, 4, 5, 0]
sole.duplicateZeros(arr)
self.assertEqual([1, 0, 0, 2, 3, 0, 0, 4], arr)
def test_mctFromLeafValues(self):
sole = ArraySolution()
self.assertEqual(32, sole.mctFromLeafValues([6, 2, 4]))
def test_matrixScore(self):
sole = ArraySolution()
self.assertEqual(
39, sole.matrixScore([[0, 0, 1, 1], [1, 0, 1, 0], [1, 1, 0, 0]]))
def test_sortArray(self):
sole = ArraySolution()
self.assertEqual([1, 2, 3, 5], sole.sortArray([5, 2, 3, 1]))
self.assertEqual([0, 0, 1, 1, 2, 5],
sole.sortArray([5, 1, 1, 2, 0, 0], 3))
def test_subarraysDivByK(self):
sole = ArraySolution()
self.assertEqual(7, sole.subarraysDivByK([4, 5, 0, -2, -3, 1], 5))
def test_relativeSortArray(self):
sole = ArraySolution()
self.assertEqual(
[2, 2, 2, 1, 4, 3, 3, 9, 6, 7, 19],
sole.relativeSortArray([2, 3, 1, 3, 2, 4, 6, 7, 9, 2, 19],
[2, 1, 4, 3, 9, 6]))
def test_findPoisonedDuration(self):
solu = ArraySolution()
self.assertEqual(4, solu.findPoisonedDuration([1, 4], 2))
self.assertEqual(3, solu.findPoisonedDuration([1, 2], 2))
def test_rob(self):
sole = ArraySolution()
self.assertEqual(4, sole.rob([1, 2, 3, 1]))
def test_intervalIntersection(self):
sole = ArraySolution()
self.assertEqual(
[[1, 2], [5, 5], [8, 10], [15, 23], [24, 24], [25, 25]],
sole.intervalIntersection(A=[[0, 2], [5, 10], [13, 23], [24, 25]],
B=[[1, 5], [8, 12], [15, 24], [25, 26]]))
def test_new21Game(self):
sole = ArraySolution()
self.assertEqual(0.6, sole.new21Game(6, 1, 10))
self.assertEqual(0.73278, sole.new21Game(21, 17, 10))
def test_pancakeSort(self):
sole = ArraySolution()
print(sole.pancakeSort([3, 2, 4, 1]))
def test_productExceptSelf(self):
sole = ArraySolution()
self.assertEqual([24, 12, 8, 6], sole.productExceptSelf([1, 2, 3, 4]))
self.assertEqual([0, -18, 0], sole.productExceptSelf([9, 0, -2]))
def test_stoneGame(self):
sole = ArraySolution()
self.assertTrue(sole.stoneGame([5, 3, 4, 5]))
def test_searchInsert(self):
sole = ArraySolution()
self.assertEqual(2, sole.searchInsert([1, 3, 5, 6], 5))
self.assertEqual(1, sole.searchInsert([1, 3, 5, 6], 2))
self.assertEqual(4, sole.searchInsert([1, 3, 5, 6], 7))
def test_mctFromLeafValues(self):
sole = ArraySolution()
# self.assertEqual(32,sole.mctFromLeafValues([6,2,4]))
self.assertEqual(644,
sole.mctFromLeafValues([9, 14, 6, 4, 13, 12, 1, 6]))
def test_maxEqualRowsAfterFlips(self):
sole = ArraySolution()
self.assertEqual(1, sole.maxEqualRowsAfterFlips([[0, 1], [1, 1]]))
self.assertEqual(
2, sole.maxEqualRowsAfterFlips([[0, 0, 0], [0, 0, 1], [1, 1, 0]]))