結果

問題 No.990 N×Mマス計算(Kの倍数)
ユーザー prd_xxx
提出日時 2020-02-14 22:54:32
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 868 bytes
コンパイル時間 251 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 47,688 KB
最終ジャッジ日時 2024-11-16 01:17:21
合計ジャッジ時間 11,755 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 16 TLE * 3
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
input = sys.stdin.readline
N,M,K = map(int,input().split())
op,*B = input().split()
A = [int(input()) for i in range(N)]
B = [int(b) for b in B]

from collections import Counter
ans = 0
if op=='+':
    c = Counter()
    for a in A:
        c[a%K] += 1
    for b in B:
        ans += c[(K-b%K)%K]
else:
    ds = set()
    m = 1
    while m*m <= K:
        if K%m==0:
            ds.add(m)
            ds.add(K//m)
        m += 1
    ds = sorted(list(ds), reverse=True)
    ca = Counter()
    for a in A:
        for d in ds:
            if a%d==0:
                ca[d] += 1
                break
    cb = Counter()
    for b in B:
        for d in ds:
            if b%d==0:
                cb[d] += 1
                break
    for ka,va in ca.items():
        for kb,vb in cb.items():
            if (ka*kb)%K == 0:
                ans += va*vb

print(ans)
0