結果

問題 No.990 N×Mマス計算(Kの倍数)
ユーザー beet
提出日時 2020-02-14 22:13:40
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 363 ms / 2,000 ms
コード長 877 bytes
コンパイル時間 2,868 ms
コンパイル使用メモリ 202,712 KB
最終ジャッジ日時 2025-01-09 00:17:45
ジャッジサーバーID
(参考情報)
judge4 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>
using namespace std;
using Int = long long;
template<typename T1,typename T2> inline void chmin(T1 &a,T2 b){if(a>b) a=b;}
template<typename T1,typename T2> inline void chmax(T1 &a,T2 b){if(a<b) a=b;}

//INSERT ABOVE HERE
signed main(){
  Int n,m,k;
  cin>>n>>m>>k;
  char op;
  cin>>op;
  vector<Int> as(n),bs(m);
  for(Int i=0;i<m;i++) cin>>bs[i];
  for(Int i=0;i<n;i++) cin>>as[i];

  if(op=='+'){
    Int ans=0;
    map<Int, Int> ma;
    for(Int a:as) ma[(k-(a%k))%k]++;
    for(Int b:bs) ans+=ma[b%k];
    cout<<ans<<endl;
  }

  if(op=='*'){
    for(Int &a:as) a=__gcd(a,k);
    for(Int &b:bs) b=__gcd(b,k);
    map<Int, Int> ma,mb;
    for(Int a:as) ma[a]++;
    for(Int b:bs) mb[b]++;
    Int ans=0;
    for(auto p:ma)
      for(auto q:mb)
        if(p.first*q.first%k==0)
          ans+=p.second*q.second;
    cout<<ans<<endl;
  }
  return 0;
}
0