From 51cf8ff6373b070660c42bcde813389156ebea1a Mon Sep 17 00:00:00 2001 From: yenru0 Date: Thu, 9 Jan 2025 19:15:32 +0900 Subject: [PATCH] complete 7453.py --- zeta_python/completed/7453.py | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 zeta_python/completed/7453.py diff --git a/zeta_python/completed/7453.py b/zeta_python/completed/7453.py new file mode 100644 index 0000000..f0f8580 --- /dev/null +++ b/zeta_python/completed/7453.py @@ -0,0 +1,34 @@ +import sys + +input = sys.stdin.readline + + +class FourArray: + def __init__(self, N, mat): + self.__N = N + self.__arr: list[list[int]] = mat # shape = (4, N) + + self.__firsts = dict() + for i in range(self.__N): + for j in range(self.__N): + first = self.__arr[i][0] + self.__arr[j][1] + if first in self.__firsts: + self.__firsts[first] += 1 + else: + self.__firsts[first] = 1 + + def solve(self) -> int: + s = 0 + for i in range(self.__N): + for j in range(self.__N): + target = -(self.__arr[i][2] + self.__arr[j][3]) + if target in self.__firsts: + s += self.__firsts[target] + + return s + + +if __name__ == "__main__": + N = int(input()) + solver = FourArray(N, [list(map(int, input().split())) for _ in range(N)]) + print(solver.solve())