結果

問題 No.990 N×Mマス計算（Kの倍数）
ユーザー mkawa2
提出日時 2021-04-06 01:15:35
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,552 ms / 2,000 ms
コード長 1,411 bytes
コンパイル時間 239 ms
コンパイル使用メモリ 82,732 KB
実行使用メモリ 111,060 KB
最終ジャッジ日時 2025-01-02 03:40:33
合計ジャッジ時間 6,163 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

sys.setrecursionlimit(10**6)
int1 = lambda x: int(x)-1
p2D = lambda x: print(*x, sep="\n")
def II(): return int(sys.stdin.buffer.readline())
def LI(): return list(map(int, sys.stdin.buffer.readline().split()))
def LI1(): return list(map(int1, sys.stdin.buffer.readline().split()))
def LLI(rows_number): return [LI() for _ in range(rows_number)]
def LLI1(rows_number): return [LI1() for _ in range(rows_number)]
def BI(): return sys.stdin.buffer.readline().rstrip()
def SI(): return sys.stdin.buffer.readline().rstrip().decode()
# dij = [(0, 1), (-1, 0), (0, -1), (1, 0)]
# dij = [(0, 1), (-1, 0), (0, -1), (1, 0), (1, 1), (1, -1), (-1, 1), (-1, -1)]
dij = [(0, 1), (1, 0), (1, 1), (1, -1)]
inf = 10**19
# md = 998244353
md = 10**9+7

def factors(a):
    if a <= 0: return [0]
    dd0, dd1 = [], []
    for d in range(1, a+1):
        if d*d > a: break
        if a%d: continue
        dd0.append(d)
        dd1.append(a//d)
    if dd0[-1] == dd1[-1]: dd1.pop()
    return dd0+dd1[::-1]

from collections import defaultdict
from math import gcd

h, w, k = LI()
op, *bb = SI().split()
bb = list(map(int, bb))
aa = [II() for _ in range(h)]

ans = 0
cnt = defaultdict(int)
if op == "+":
    for b in bb: cnt[b%k] += 1
    for a in aa: ans += cnt[(k-a%k)%k]
else:
    for b in bb:
        for f in factors(gcd(b, k)):
            cnt[f] += 1
    for a in aa:
        ans += cnt[k//gcd(a, k)]

print(ans)
0