結果

問題 No.990 N×Mマス計算（Kの倍数）
コンテスト
ユーザー brthyyjp
提出日時 2020-02-15 00:53:41
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 199 ms / 2,000 ms
コード長 1,105 bytes
コンパイル時間 270 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 112,176 KB
最終ジャッジ日時 2024-11-16 02:07:55
合計ジャッジ時間 3,280 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

n, m, k = map(int, input().split())
l = list(map(str, input().split()))
b = [int(l[i]) for i in range(1, m+1)]

op = l[0]
#print(op)
#rint(b)

a = [0]*n
for i in range(n):
    a[i] = int(input())

if op == '+':
    da = {}
    for i in range(n):
        if a[i]%k not in da:
            da[a[i]%k] = 1
        else:
            da[a[i]%k] += 1

    db = {}
    for i in range(m):
        if b[i]%k not in db:
            db[b[i]%k] = 1
        else:
            db[b[i]%k] += 1
    ans = 0
    for i in da:
        if i == 0:
            j = 0
        else:
            j = k-i
        if j in db:
            ans += da[i]*db[j]

else:
    import math
    da = {}
    for i in range(n):
        g = math.gcd(a[i], k)
        if g not in da:
            da[g] = 1
        else:
            da[g] += 1

    db = {}
    for i in range(m):
        g = math.gcd(b[i], k)
        if g not in db:
            db[g] = 1
        else:
            db[g] += 1
    ans = 0
    for i in da:
        for j in db:
            if (i*j)%k == 0:
                ans += da[i]*db[j]
    #print(da)
    #print(db)

print(ans)
0