Exemplo n.º 1
0
 def test_word_squares(self):
     self.assertEqual([['wall', 'area', 'lead', 'lady'], ['ball', 'area', 'lead', 'lady']], \
                      word_squares(["area", "lead", "wall", "lady", "ball"]))
# l a d y
# Note:
# There are at least 1 and at most 1000 words.
# All words will have the exact same length.
# Word length is at least 1 and at most 5.
# Each word contains only lowercase English alphabet a-z.

# Example 1:

# Input:
# ["area","lead","wall","lady","ball"]

# Output:
# [
# [ "wall",
# "area",
# "lead",
# "lady"
# ],
# [ "ball",
# "area",
# "lead",
# "lady"
# ]
# ]
from algorithms.strings import word_squares

a = ["area", "lead", "wall", "lady", "ball"]

print(word_squares(a))
Exemplo n.º 3
0
 def test_word_squares(self):
     self.assertEqual([['wall', 'area', 'lead', 'lady'], ['ball', 'area', 'lead', 'lady']], \
                      word_squares(["area", "lead", "wall", "lady", "ball"]))
 def test_word_squares(self):
     self.assertEqual(
         [["wall", "area", "lead", "lady"],
          ["ball", "area", "lead", "lady"]],
         word_squares(["area", "lead", "wall", "lady", "ball"]),
     )