結果
問題 |
No.990 N×Mマス計算(Kの倍数)
|
ユーザー |
|
提出日時 | 2022-07-09 15:15:28 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 897 bytes |
コンパイル時間 | 145 ms |
コンパイル使用メモリ | 82,364 KB |
実行使用メモリ | 92,704 KB |
最終ジャッジ日時 | 2024-12-31 00:58:11 |
合計ジャッジ時間 | 7,946 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 17 TLE * 2 |
ソースコード
from math import gcd n, m, k = map(int, input().split()) op, *B = input().split() B = [int(b) for b in B] A = [int(input()) for _ in range(n)] if op == "+": cnt = {} for b in B: cnt[b % k] = cnt.get(b % k, 0) + 1 ans = 0 for a in A: ans += cnt.get(-a % k, 0) print(ans) else: div = [] for i in range(1, int(k ** 0.5 + 1)): if k % i == 0: div.append(i) if i * i != k: div.append(k // i) else: pass else: pass cnt = {} div.sort() for b in B: g = gcd(b, k) for d in div: if d > b: break elif g % d == 0: cnt[d] = cnt.get(d, 0) + 1 else: pass ans = 0 for a in A: g = gcd(a, k) ans += cnt.get(k // g, 0) print(ans)