#426392 (C++17) No.990 N×Mマス計算（Kの倍数）

• 問題: No.990 N×Mマス計算（Kの倍数）
• ユーザー: %20
• 提出日時: 2020-02-05 18:24:01
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 87 ms / 2,000 ms
• コード長: 1,227 bytes
• コンパイル時間: 1,920 ms
• コンパイル使用メモリ: 217,612 KB
• 実行使用メモリ: 14,208 KB
• 最終ジャッジ日時: 2024-04-27 19:26:57
• 合計ジャッジ時間: 3,002 ms
• ジャッジサーバーID （参考情報）: judge4 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 6,816 KB
testcase_01	AC	1 ms; 6,940 KB
testcase_02	AC	1 ms; 6,940 KB
testcase_03	AC	2 ms; 6,940 KB
testcase_04	AC	2 ms; 6,940 KB
testcase_05	AC	2 ms; 6,940 KB
testcase_06	AC	1 ms; 6,944 KB
testcase_07	AC	2 ms; 6,940 KB
testcase_08	AC	1 ms; 6,940 KB
testcase_09	AC	2 ms; 6,940 KB
testcase_10	AC	27 ms; 6,944 KB
testcase_11	AC	7 ms; 6,940 KB
testcase_12	AC	72 ms; 6,944 KB
testcase_13	AC	7 ms; 6,940 KB
testcase_14	AC	20 ms; 6,940 KB
testcase_15	AC	10 ms; 6,940 KB
testcase_16	AC	32 ms; 7,680 KB
testcase_17	AC	9 ms; 6,940 KB
testcase_18	AC	73 ms; 6,944 KB
testcase_19	AC	23 ms; 6,940 KB
testcase_20	AC	87 ms; 14,208 KB

ソースコード

#include<bits/stdc++.h>
using namespace std;

using ll=long long;
const int e5=1e5;
const int e9=1e9;

void readnum(int&n,int l,char s){
	int c=getchar();
	assert('1'<=c&&c<='9');
	n=c-'0';
	while(true){
		c=getchar();
		if(c==s)break;
		assert('0'<=c&&c<='9');
		assert(n*10ll+c-'0'<=l);
		n=n*10+c-'0';
	}
}

int gcd(int a,int b){
	return a?gcd(b%a,a):b;
}
ll solve3(int N,int M,int K,int op,vector<int>A,vector<int>B){
	ll z=0;
	if(op=='+'){
		map<int,int>m;
		for(int i=0;i<M;++i){
			++m[B[i]%K];
		}
		for(int i=0;i<N;++i){
			z+=m[(K-A[i]%K)%K];
		}
	}else{
		map<ll,ll>ma,mb;
		for(int i=0;i<N;++i){
			++ma[gcd(K,A[i])];
		}
		for(int i=0;i<M;++i){
			++mb[gcd(K,B[i])];
		}
		for(auto a:ma){
			for(auto b:mb){
				if(a.first*b.first%K==0){
					z+=a.second*b.second;
				}
			}
		}
	}
	return z;
}

int main(){cin.tie(0);ios::sync_with_stdio(false);
	int N,M,K;
	readnum(N,e5,' ');
	readnum(M,e5,' ');
	readnum(K,e9,'\n');
	int op=getchar();
	assert(op=='+'||op=='*');
	int c=getchar();
	assert(c==' ');
	vector<int>A(N),B(M);
	for(int i=0;i<M;++i){
		readnum(B[i],e9," \n"[i==M-1]);
	}
	for(int i=0;i<N;++i){
		readnum(A[i],e9,'\n');
	}
	c=getchar();
	assert(c==-1);
	cout<<solve3(N,M,K,op,A,B)<<"\n";
	return 0;
}