結果

問題 No.990 N×Mマス計算（Kの倍数）
コンテスト
ユーザー aaaaaaaaaa2230
提出日時 2022-04-16 10:50:11
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 349 ms / 2,000 ms
コード長 812 bytes
コンパイル時間 240 ms
コンパイル使用メモリ 82,008 KB
実行使用メモリ 97,448 KB
最終ジャッジ日時 2024-12-25 15:37:53
合計ジャッジ時間 3,775 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

import math
n,m,k = map(int,input().split())
L = list(input().split())
op = L[0]
B = [int(b) for b in L[1:]]
A = [int(input()) for i in range(n)] 

if op == "+":
    dic = {}
    for b in B:
        dic[b%k] = dic.get(b%k,0)+1

    ans = 0
    for a in A:
        ans += dic.get((k-a)%k,0)
    
else:
    dicB = {}

    for b in B:
        g = math.gcd(b,k)
        dicB[g] = dicB.get(g,0)+1

    dicA = {}

    for a in A:
        g = math.gcd(a,k)
        dicA[g] = dicA.get(g,0)+1

    divs = set()
    for i in range(1,int(k**0.5)+1):
        if k%i:
            continue
        divs.add(i)
        divs.add(k//i)

    ans = 0
    for i in divs:
        for j in divs:
            x = i*j
            if math.gcd(x,k) != k:
                continue
            ans += dicA.get(i,0)*dicB.get(j,0)
print(ans)
0