diff --git a/technical_tests/sand_grain_drop.py b/technical_tests/sand_grain_drop.py index 93d5284..ab34797 100644 --- a/technical_tests/sand_grain_drop.py +++ b/technical_tests/sand_grain_drop.py @@ -37,7 +37,14 @@ def main(pile, n): center = int((len(pile) - 1) / 2) - return center + while n != 0: + + if pile[center][center] < 3: + pile[center][center] += 1 + + n -= 1 + + return pile if __name__ == "__main__":