結果

問題 No.990 N×Mマス計算（Kの倍数）
ユーザー ntuda
提出日時 2025-01-25 09:34:59
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 874 bytes
コンパイル時間 998 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 99,712 KB
最終ジャッジ日時 2025-01-25 09:35:11
合計ジャッジ時間 11,969 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 17 TLE * 2
権限があれば一括ダウンロードができます

ソースコード

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)
N, M, K = map(int, input().split())
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:
        for d in D:
            if b % d == 0:
                dic[d] += 1
    for a in A:
        x = K // gcd(a,K)
        if x in dic:
            ans += dic[x]
print(ans)
0