結果

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

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
#include <unordered_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);
      }
      vector<long long> C;
      for(long long i = 1; k >= i*i; i++){
        if(!(k%i)){
          C.push_back(i);
          if(i*i != k)C.push_back(k/i);
        }
      }
      unordered_map<int,int> nya;
      for(int i = 0; m > i; i++){
        for(int j = 0; C.size() > j; j++){
          if(!(B[i]%C[j])){
            nya[C[j]]++;
          }
        }
      }
      for(int i = 0; n > i; i++){
        ans += nya[k/A[i]];
      }
      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