diff options
Diffstat (limited to 'challenge-337/pokgopun/python/ch-2.py')
| -rw-r--r-- | challenge-337/pokgopun/python/ch-2.py | 273 |
1 files changed, 273 insertions, 0 deletions
diff --git a/challenge-337/pokgopun/python/ch-2.py b/challenge-337/pokgopun/python/ch-2.py new file mode 100644 index 0000000000..1aaef66a6d --- /dev/null +++ b/challenge-337/pokgopun/python/ch-2.py @@ -0,0 +1,273 @@ +### https://theweeklychallenge.org/blog/perl-weekly-challenge-337/ +""" + +Task 2: Odd Matrix + +Submitted by: [41]Mohammad Sajid Anwar + __________________________________________________________________ + + You are given row and col, also a list of positions in the matrix. + + Write a script to perform action on each location (0-indexed) as + provided in the list and find out the total odd valued cells. + + For each location (r, c), do both of the following: +a) Increment by 1 all the cells on row r. +b) Increment by 1 all the cells on column c. + +Example 1 +Input: $row = 2, $col = 3, @locations = ([0,1],[1,1]) +Output: 6 + +Initial: +[ 0 0 0 ] +[ 0 0 0 ] + +Apply [0,1]: +Increment row 0: +Before After +[ 0 0 0 ] [ 1 1 1 ] +[ 0 0 0 ] [ 0 0 0 ] +Increment col 1: +Before After +[ 1 1 1 ] [ 1 2 1 ] +[ 0 0 0 ] [ 0 1 0 ] + +Apply [1,1]: +Increment row 1: +Before After +[ 1 2 1 ] [ 1 2 1 ] +[ 0 1 0 ] [ 1 2 1 ] +Increment col 1: +Before After +[ 1 2 1 ] [ 1 3 1 ] +[ 1 2 1 ] [ 1 3 1 ] + +Final: +[ 1 3 1 ] +[ 1 3 1 ] + +Example 2 +Input: $row = 2, $col = 2, @locations = ([1,1],[0,0]) +Output: 0 + +Initial: +[ 0 0 ] +[ 0 0 ] + +Apply [1,1]: +Increment row 1: +Before After +[ 0 0 ] [ 0 0 ] +[ 0 0 ] [ 1 1 ] +Increment col 1: +Before After +[ 0 0 ] [ 0 1 ] +[ 1 1 ] [ 1 2 ] + +Apply [0,0]: +Increment row 0: +Before After +[ 0 1 ] [ 1 2 ] +[ 1 2 ] [ 1 2 ] +Increment col 0: +Before After +[ 1 2 ] [ 2 2 ] +[ 1 2 ] [ 2 2 ] + +Final: +[ 2 2 ] +[ 2 2 ] + +Example 3 +Input: $row = 3, $col = 3, @locations = ([0,0],[1,2],[2,1]) +Output: 0 + +Initial: +[ 0 0 0 ] +[ 0 0 0 ] +[ 0 0 0 ] + +Apply [0,0]: +Increment row 0: +Before After +[ 0 0 0 ] [ 1 1 1 ] +[ 0 0 0 ] [ 0 0 0 ] +[ 0 0 0 ] [ 0 0 0 ] +Increment col 0: +Before After +[ 1 1 1 ] [ 2 1 1 ] +[ 0 0 0 ] [ 1 0 0 ] +[ 0 0 0 ] [ 1 0 0 ] + +Apply [1,2]: +Increment row 1: +Before After +[ 2 1 1 ] [ 2 1 1 ] +[ 1 0 0 ] [ 2 1 1 ] +[ 1 0 0 ] [ 1 0 0 ] +Increment col 2: +Before After +[ 2 1 1 ] [ 2 1 2 ] +[ 2 1 1 ] [ 2 1 2 ] +[ 1 0 0 ] [ 1 0 1 ] + +Apply [2,1]: +Increment row 2: +Before After +[ 2 1 2 ] [ 2 1 2 ] +[ 2 1 2 ] [ 2 1 2 ] +[ 1 0 1 ] [ 2 1 2 ] +Increment col 1: +Before After +[ 2 1 2 ] [ 2 2 2 ] +[ 2 1 2 ] [ 2 2 2 ] +[ 2 1 2 ] [ 2 2 2 ] + +Final: +[ 2 2 2 ] +[ 2 2 2 ] +[ 2 2 2 ] + +Example 4 +Input: $row = 1, $col = 5, @locations = ([0,2],[0,4]) +Output: 2 + +Initial: +[ 0 0 0 0 0 ] + +Apply [0,2]: +Increment row 0: +Before After +[ 0 0 0 0 0 ] [ 1 1 1 1 1 ] +Increment col 2: +Before After +[ 1 1 1 1 1 ] [ 1 1 2 1 1 ] + +Apply [0,4]: +Increment row 0: +Before After +[ 1 1 2 1 1 ] [ 2 2 3 2 2 ] +Increment col 4: +Before After +[ 2 2 3 2 2 ] [ 2 2 3 2 3 ] + +Final: +[ 2 2 3 2 3 ] + +Example 5 +Input: $row = 4, $col = 2, @locations = ([1,0],[3,1],[2,0],[0,1]) +Output: 8 + +Initial: +[ 0 0 ] +[ 0 0 ] +[ 0 0 ] +[ 0 0 ] + +Apply [1,0]: +Increment row 1: +Before After +[ 0 0 ] [ 0 0 ] +[ 0 0 ] [ 1 1 ] +[ 0 0 ] [ 0 0 ] +[ 0 0 ] [ 0 0 ] +Increment col 0: +Before After +[ 0 0 ] [ 1 0 ] +[ 1 1 ] [ 2 1 ] +[ 0 0 ] [ 1 0 ] +[ 0 0 ] [ 1 0 ] + +Apply [3,1]: +Increment row 3: +Before After +[ 1 0 ] [ 1 0 ] +[ 2 1 ] [ 2 1 ] +[ 1 0 ] [ 1 0 ] +[ 1 0 ] [ 2 1 ] +Increment col 1: +Before After +[ 1 0 ] [ 1 1 ] +[ 2 1 ] [ 2 2 ] +[ 1 0 ] [ 1 1 ] +[ 2 1 ] [ 2 2 ] + +Apply [2,0]: +Increment row 2: +Before After +[ 1 1 ] [ 1 1 ] +[ 2 2 ] [ 2 2 ] +[ 1 1 ] [ 2 2 ] +[ 2 2 ] [ 2 2 ] +Increment col 0: +Before After +[ 1 1 ] [ 2 1 ] +[ 2 2 ] [ 3 2 ] +[ 2 2 ] [ 3 2 ] +[ 2 2 ] [ 3 2 ] + +Apply [0,1]: +Increment row 0: +Before After +[ 2 1 ] [ 3 2 ] +[ 3 2 ] [ 3 2 ] +[ 3 2 ] [ 3 2 ] +[ 3 2 ] [ 3 2 ] +Increment col 1: +Before After +[ 3 2 ] [ 3 3 ] +[ 3 2 ] [ 3 3 ] +[ 3 2 ] [ 3 3 ] +[ 3 2 ] [ 3 3 ] + +Final: +[ 3 3 ] +[ 3 3 ] +[ 3 3 ] +[ 3 3 ] + __________________________________________________________________ + + Last date to submit the solution 23:59 (UK Time) Sunday 7th September + 2025. + __________________________________________________________________ + +SO WHAT DO YOU THINK ? +""" +### solution by pokgopun@gmail.com + +from dataclasses import dataclass + +@dataclass +class Position: + r: int + c: int + +def om(row: int, col: int, positions: tuple[Position]) -> int: + p2c: dict[int,int] = {} + for p in positions: + p2c[p.r] = p2c.get(p.r,0) + 1 + p2c[-1-p.c] = p2c.get(-1-p.c,0) + 1 + #print(p2c) + cnto = 0 + for r in range(row): + for c in range(col): + if (p2c.get(r,0)+p2c.get(-1-c,0)) % 2 == 1: + cnto += 1 + return cnto + +import unittest + +class TestOm(unittest.TestCase): + def test(self): + for (row, col, locations), otpt in { + (2, 3, ((0,1),(1,1))): 6, + (2, 2, ((1,1),(0,0))): 0, + (3, 3, ((0,0),(1,2),(2,1))): 0, + (1, 5, ((0,2),(0,4))): 2, + (4, 2, ((1,0),(3,1),(2,0),(0,1))): 8, + }.items(): + #print(row, col, locations, otpt) + self.assertEqual(om(row, col, tuple(Position(e[0],e[1]) for e in locations)), otpt) + +unittest.main() |
