def test_3x4(self):
grid1 = [["0", "E", "0", "0"], ["E", "0", "W", "E"],
["0", "E", "0", "0"]]
self.assertEqual(3, bomb_enemy.max_killed_enemies(grid1))
grid1 = [["0", "E", "0", "E"], ["E", "E", "E", "0"],
["E", "0", "W", "E"], ["0", "E", "0", "0"]]
grid2 = [["0", "0", "0", "E"], ["E", "0", "0", "0"],
["E", "0", "W", "E"], ["0", "E", "0", "0"]]
self.assertEqual(5, bomb_enemy.max_killed_enemies(grid1))
self.assertEqual(3, bomb_enemy.max_killed_enemies(grid2))
def test_3x4(self):
grid1 = [
["0","E","0","0"],
["E","0","W","E"],
["0","E","0","0"]
]
self.assertEqual(3, bomb_enemy.max_killed_enemies(grid1))
grid1 = [
["0", "E", "0", "E"],
["E", "E", "E", "0"],
["E", "0", "W", "E"],
["0", "E", "0", "0"]
]
grid2 = [
["0", "0", "0", "E"],
["E", "0", "0", "0"],
["E", "0", "W", "E"],
["0", "E", "0", "0"]
]
self.assertEqual(5, bomb_enemy.max_killed_enemies(grid1))
self.assertEqual(3, bomb_enemy.max_killed_enemies(grid2))
"""
Given a 2D grid, each cell is either a wall 'W',
an enemy 'E' or empty '0' (the number zero),
return the maximum enemies you can kill using one bomb.
The bomb kills all the enemies in the same row and column from
the planted point until it hits the wall since the wall is too strong
to be destroyed.
Note that you can only put the bomb at an empty cell.
Example:
For the given grid
0 E 0 0
E 0 W E
0 E 0 0
return 3. (Placing a bomb at (1,1) kills 3 enemies)
"""
from algorithms.matrix.bomb_enemy import max_killed_enemies
grid1 = [["0", "E", "0", "0"], ["E", "0", "W", "E"], ["0", "E", "0", "0"]]
grid2 = [["0", "E", "0", "E"], ["E", "E", "E", "0"], ["E", "0", "W", "E"],
["0", "E", "0", "0"]]
grid3 = [["0", "0", "0", "E"], ["E", "0", "0", "0"], ["E", "0", "W", "E"],
["0", "E", "0", "0"]]
print(max_killed_enemies(grid1))
print(max_killed_enemies(grid2))
print(max_killed_enemies(grid3))