結果
問題 |
No.990 N×Mマス計算(Kの倍数)
|
ユーザー |
|
提出日時 | 2020-02-14 23:30:56 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 784 ms / 2,000 ms |
コード長 | 589 bytes |
コンパイル時間 | 247 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 38,360 KB |
最終ジャッジ日時 | 2024-11-16 01:42:21 |
合計ジャッジ時間 | 5,029 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 19 |
ソースコード
import math from collections import Counter n, m, k = map(int, input().split()) line = input().split() op = line[0] B = list(map(int, line[1:])) A = [int(input()) for _ in range(n)] if op == '+': mod = [a % k for a in A] modcnt = Counter(mod) ans = sum([modcnt[(k - (b%k)) % k] for b in B]) else: gcda_cnt = Counter([math.gcd(k,a) for a in A]) gcdb_cnt = Counter([math.gcd(k,b) for b in B]) ans = 0 for ga, gacnt in gcda_cnt.items(): for gb, gbcnt in gcdb_cnt.items(): if (ga * gb) % k == 0: ans += gacnt * gbcnt print(ans)