結果

問題 No.990 N×Mマス計算（Kの倍数）
ユーザー kotatsugamekotatsugame
提出日時 2020-02-07 04:40:10
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 213 ms / 2,000 ms
コード長 692 bytes
コンパイル時間 827 ms
コンパイル使用メモリ 78,720 KB
実行使用メモリ 14,592 KB
最終ジャッジ日時 2024-11-16 00:27:01
合計ジャッジ時間 2,578 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 19
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:10:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   10 | main()
      | ^~~~

ソースコード

diff # 

#include<iostream>
#include<map>
using namespace std;
int N,M;
char op;
long A[1<<17],B[1<<17];
long K,ans;
map<int,int>X;
int gcd(int a,int b){return b?gcd(b,a%b):a;}
main()
{
	cin>>N>>M>>K;
	cin>>op;
	for(int j=0;j<M;j++)
	{
		cin>>B[j];
		X[op=='+'?B[j]%K:gcd(B[j],K)]++;
	}
	if(op=='*')
	{
		for(int i=1;i*i<=K;i++)
		{
			if(K%i==0)X[i]+=0,X[K/i]+=0;
		}
		for(map<int,int>::iterator it=X.begin();it!=X.end();it++)
		{
			int k=it->first,v=it->second;
			for(int i=1;i*i<=k;i++)
			{
				if(k%i==0)
				{
					if(i!=k)X[i]+=v;
					if(k/i!=i&&i!=1)X[k/i]+=v;
				}
			}
		}
	}
	for(int i=0;i<N;i++)
	{
		cin>>A[i];
		ans+=op=='+'?X[(K-A[i]%K)%K]:X[K/gcd(A[i],K)];
	}
	cout<<ans<<endl;
}
0