結果

問題 No.990 N×Mマス計算（Kの倍数）
ユーザー aaaaaaaaaa2230aaaaaaaaaa2230
提出日時 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 33 ms
51,968 KB
testcase_01 AC 35 ms
52,224 KB
testcase_02 AC 37 ms
52,480 KB
testcase_03 AC 37 ms
51,968 KB
testcase_04 AC 39 ms
58,624 KB
testcase_05 AC 35 ms
52,096 KB
testcase_06 AC 36 ms
52,224 KB
testcase_07 AC 39 ms
52,352 KB
testcase_08 AC 35 ms
52,864 KB
testcase_09 AC 34 ms
52,352 KB
testcase_10 AC 95 ms
85,080 KB
testcase_11 AC 112 ms
88,832 KB
testcase_12 AC 349 ms
90,988 KB
testcase_13 AC 96 ms
84,712 KB
testcase_14 AC 113 ms
85,772 KB
testcase_15 AC 91 ms
79,872 KB
testcase_16 AC 101 ms
84,528 KB
testcase_17 AC 84 ms
79,744 KB
testcase_18 AC 346 ms
90,768 KB
testcase_19 AC 262 ms
83,784 KB
testcase_20 AC 144 ms
97,448 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