파이썬 | 백준 | 7453 | 합이 0인 네 정수

 

Solution. dictionary

# 7453, 합이 0인 네 정수
import sys
from collections import defaultdict

A, B, C, D = [], [], [], []
N = int(sys.stdin.readline())
for i in range(N):
    a, b, c, d = map(int, sys.stdin.readline().split())
    A.append(a)
    B.append(b)
    C.append(c)
    D.append(d)

# abPlus = defaultdict(int)
# for a in A:
#     for b in B:
#         abPlus[a+b] = abPlus[a+b]+1
#
# count = 0
# for c in C:
#     for d in D:
#         count += abPlus[-(c+d)]

abPlus = dict()
for a in A:
    for b in B:
        abPlus[a + b] = abPlus.get(a + b, 0) + 1
count = 0
for c in C:
    for d in D:
        count += abPlus.get(-(c + d), 0)

print(count)

defaultdict: 시간초과

 

www.acmicpc.net/problem/7453

7453번: 합이 0인 네 정수