diff options
Diffstat (limited to 'challenge-152/lubos-kolouch/python/ch-2.py')
| -rw-r--r-- | challenge-152/lubos-kolouch/python/ch-2.py | 43 |
1 files changed, 43 insertions, 0 deletions
diff --git a/challenge-152/lubos-kolouch/python/ch-2.py b/challenge-152/lubos-kolouch/python/ch-2.py new file mode 100644 index 0000000000..a7fe673bd6 --- /dev/null +++ b/challenge-152/lubos-kolouch/python/ch-2.py @@ -0,0 +1,43 @@ +class Point: + def __init__(self, x: int, y: int): + self.x = x + self.y = y + + +class Rectangle: + def __init__(self, left: Point, right: Point): + self.left = left + self.right = right + + def get_area(self) -> int: + return abs(self.left.x - self.right.x) * abs(self.left.y - + self.right.y) + + +def get_total_area(first: Rectangle, second: Rectangle) -> int: + area1 = first.get_area() + area2 = second.get_area() + + # calculate the overlapping area + x_dist = min(first.right.x, second.right.x) - \ + max(first.left.x, second.left.x) + y_dist = min(first.right.y, second.right.y) - \ + max(first.left.y, second.left.y) + area_i = 0 + if x_dist > 0 and y_dist > 0: + area_i = x_dist * y_dist + + return area1 + area2 - area_i + + +first = Rectangle(Point(-1, 0), Point(2, 2)) +second = Rectangle(Point(0, -1), Point(4, 4)) + +if get_total_area(first, second) != 22: + raise Exception("Failed test 1") + +first = Rectangle(Point(-3, -1), Point(1, 3)) +second = Rectangle(Point(-1, -3), Point(2, 2)) + +if get_total_area(first, second) != 25: + raise Exception("Failed test 2") |
