結果

問題 No.990 N×Mマス計算（Kの倍数）
コンテスト
ユーザー %20
提出日時 2020-02-07 00:21:04
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 865 ms / 2,000 ms
コード長 506 bytes
コンパイル時間 187 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 33,384 KB
最終ジャッジ日時 2024-11-16 00:26:44
合計ジャッジ時間 5,300 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

from math import gcd
N,M,K=map(int,input().split())
op,*B=input().split()
B=map(int,B)
A=[int(input())for _ in range(N)]
z=0
if op=='+':
	d={}
	for b in B:
		b%=K
		if b in d:
			d[b]+=1
		else:
			d[b]=1
	for a in A:
		a=-a%K
		if a in d:
			z+=d[a]
else:
	da={}
	db={}
	for b in B:
		g=gcd(b,K)
		if g in db:
			db[g]+=1
		else:
			db[g]=1
	for a in A:
		g=gcd(a,K)
		if g in da:
			da[g]+=1
		else:
			da[g]=1
	for ka,va in da.items():
		for kb,vb in db.items():
			if ka*kb%K==0:
				z+=va*vb
print(z)
0