結果

問題 No.990 N×Mマス計算（Kの倍数）
ユーザー ntuda
提出日時 2025-01-25 13:38:38
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 313 ms / 2,000 ms
コード長 994 bytes
コンパイル時間 274 ms
コンパイル使用メモリ 82,848 KB
実行使用メモリ 99,664 KB
最終ジャッジ日時 2025-01-25 13:40:10
合計ジャッジ時間 3,714 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge6
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict
from math import gcd

def make_divisors(n):
    divisors = []  # 必要に応じてsetにしても良いかも
    i = 1
    while i ** 2 <= n:
        if n % i == 0:
            divisors.append(i)
            if i ** 2 != n:
                divisors.append(n // i)
        i += 1
    divisors.sort()
    return divisors

dic = defaultdict(int)
dic2 = defaultdict(int)
N, M, K = map(int, input().split())
if K == 1:
    print(N * M)
    exit()
op, *B = input().split()
A = []
for i in range(N):
    A.append(int(input()))
B = [int(i) for i in B]
ans = 0
if op == "+":
    for b in B:
        dic[(-b) % K] += 1
    for a in A:
        if a % K in dic:
            ans += dic[a % K]
else:
    D = make_divisors(K)
    for b in B:
        x = gcd(b, K)
        dic[x] += 1
    for a in A:
        x = gcd(a, K)
        dic2[x] += 1
    for d1 in D:
        for d2 in D:
            if gcd(d1 * d2, K) == K:
                ans += dic[d1] * dic2[d2]
print(ans)
0