Pythonではじめる競技プログラミング 例題の解答例

abcd.py

import collections
def solve(A, B, C, D):
    count = 0
    AB = create_pairs(A, B)
    CD = create_pairs(C, D)

    for k, v in AB.items():
        count += CD[-k] * v
    return count


def create_pairs(S, T):
    pairs = collections.Counter()
    for i in S:
        for j in T:
            pairs[i + j] += 1
    return pairs