diff options
Diffstat (limited to 'challenge-068/walt-mankowski/python/ch-1.py')
| -rw-r--r-- | challenge-068/walt-mankowski/python/ch-1.py | 38 |
1 files changed, 38 insertions, 0 deletions
diff --git a/challenge-068/walt-mankowski/python/ch-1.py b/challenge-068/walt-mankowski/python/ch-1.py new file mode 100644 index 0000000000..b2c22a3923 --- /dev/null +++ b/challenge-068/walt-mankowski/python/ch-1.py @@ -0,0 +1,38 @@ +def print_mat(m): + for row in m: + print(row) + print() + +def zero_mat(m): + num_rows = len(m) + num_cols = len(m[0]) + + row_flags = [False for _ in range(num_rows)] + col_flags = [False for _ in range(num_cols)] + for r in range(num_rows): + for c in range(num_cols): + if m[r][c] == 0: + row_flags[r] = True + col_flags[c] = True + + for r in range(num_rows): + for c in range(num_cols): + if row_flags[r] or col_flags[c]: + m[r][c] = 0 + +m1 = [[1,0,1], + [1,1,1], + [1,1,1]] + +m2 = [[1,0,1], + [1,1,1], + [1,0,1]] + +print_mat(m1) +print_mat(m2) + +zero_mat(m1) +zero_mat(m2) + +print_mat(m1) +print_mat(m2) |
