結果

問題 No.990 N×Mマス計算（Kの倍数）
ユーザー aaaaaaaaaa2230aaaaaaaaaa2230
提出日時 2022-04-16 10:50:11
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 320 ms / 2,000 ms
コード長 812 bytes
コンパイル時間 438 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 97,384 KB
最終ジャッジ日時 2024-06-07 12:17:44
合計ジャッジ時間 3,935 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 40 ms
51,840 KB
testcase_01 AC 38 ms
52,480 KB
testcase_02 AC 39 ms
52,608 KB
testcase_03 AC 39 ms
52,352 KB
testcase_04 AC 42 ms
58,496 KB
testcase_05 AC 39 ms
52,224 KB
testcase_06 AC 40 ms
52,736 KB
testcase_07 AC 39 ms
52,736 KB
testcase_08 AC 39 ms
52,480 KB
testcase_09 AC 40 ms
52,352 KB
testcase_10 AC 104 ms
85,668 KB
testcase_11 AC 120 ms
88,972 KB
testcase_12 AC 318 ms
90,880 KB
testcase_13 AC 99 ms
85,248 KB
testcase_14 AC 118 ms
85,888 KB
testcase_15 AC 98 ms
79,872 KB
testcase_16 AC 117 ms
84,944 KB
testcase_17 AC 92 ms
79,616 KB
testcase_18 AC 320 ms
90,880 KB
testcase_19 AC 288 ms
83,960 KB
testcase_20 AC 189 ms
97,384 KB
権限があれば一括ダウンロードができます

ソースコード

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