結果

問題 No.990 N×Mマス計算（Kの倍数）
コンテスト
ユーザー 👑 CleyL
提出日時 2020-02-14 23:26:56
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,420 bytes
コンパイル時間 953 ms
コンパイル使用メモリ 88,288 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-16 01:41:08
合計ジャッジ時間 2,632 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 18 WA * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
#include <map>
using namespace std;
template <typename T>
T gcd(T a,T b){
  if(a%b==0)return b;
  else return gcd(b,a%b);
}
template <typename T>
T lcm(T a,T b){
  return a/gcd(a,b)*b;
}
int main(){
    long long n,m,k;cin>>n>>m>>k;
    long long ans = 0;
    char z;cin>>z;
    vector<long long> A(n);
    vector<long long> B(m);
    for(int i = 0; m > i; i++)cin>>B[i];
    for(int i = 0; n > i; i++)cin>>A[i];
    //A[i]と互いに素なB[j]の個数が答え..........O(NM)!!!!!!解散！！！！！笑
    if(z == '*'){
      for(int i = 0; m > i; i++){
        B[i] = gcd(B[i],k);
      }
      for(int i = 0; n > i; i++){
        A[i] = gcd(A[i],k);
      }
      map<long long,int> C;
      for(int i = 0; n > i; i++){
        C[A[i]]++;
      }
      map<long long,int> D;
      for(int i = 0; m > i; i++){
        D[B[i]]++;
      }
      for(auto x:C){
        for(auto y:D){
          if(!((x.first*y.first)%k)){
            ans += x.second*y.second;
          }
        }
      }
      cout << ans << endl;
    }else{//O(n log m)
      for(int i = 0; m > i; i++){
        B[i] = B[i]%k;
      }
      sort(B.begin(),B.end());
      for(int i = 0; n > i; i++){
        A[i] = A[i]%k;
        int want = (k-A[i])%k;
        ans += (upper_bound(B.begin(),B.end(),want)-lower_bound(B.begin(),B.end(),want));
      }
      cout << ans << endl;
    }
}
0