Enable Javascript in your browser and then refresh this page, for a much enhanced experience.
while + for loops solution in Clear category for Zigzag Array by kdim
from typing import List
def create_zigzag(rows: int, cols: int, start: int = 1) -> List[List[int]]:
arr, i = [], 0
while i < rows:
arr.append([i * cols + start + j for j in range(cols)][::(-1)**i])
i += 1
return arr
if __name__ == '__main__':
print("Example:")
print(create_zigzag(3, 3, 5))
assert create_zigzag(3, 5) == [
[1, 2, 3, 4, 5],
[10, 9, 8, 7, 6],
[11, 12, 13, 14, 15]
]
assert create_zigzag(5, 1) == [
[1],
[2],
[3],
[4],
[5]
]
assert create_zigzag(3, 3, 5) == [
[5, 6, 7],
[10, 9, 8],
[11, 12, 13]
]
# Edge cases
assert create_zigzag(0, 3) == []
assert create_zigzag(3, 0) == [[], [], []]
assert create_zigzag(0, 0) == []
print("Coding complete? Click 'Check' to earn cool rewards!")
Feb. 26, 2021
