結果

問題 No.990 N×Mマス計算（Kの倍数）
コンテスト
ユーザー pitsu_kyo_pro
提出日時 2020-02-15 11:15:22
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
MLE  
実行時間 -
コード長 2,116 bytes
コンパイル時間 1,022 ms
コンパイル使用メモリ 108,004 KB
実行使用メモリ 754,864 KB
最終ジャッジ日時 2024-11-16 02:09:58
合計ジャッジ時間 3,332 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 17 RE * 1 MLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<vector>
#include<algorithm>
#include<map>
#include<cstring>
#include<math.h>
#include<bitset>
#include<queue>
#include<set>
#include<iomanip>
#include<assert.h>
#include<string>
#include<cstdlib>
#include<cctype>
#include<unordered_map>
#include<sstream>
#include<numeric>
#include<stack>
// #include<bits/stdc++.h>
using namespace std;
using ll = long long;
using P = pair<int,int>;
constexpr ll INF = 9e18;
constexpr int inf = 1e9;
const double INFD = 1e100;
const ll mod = 1000000007;
const double PI = 3.1415926535897932384626433832795028841971;
const int dx[4] = {1, 0, -1, 0};
const int dy[4] = {0, 1, 0, -1};
template<class T> inline bool chmin(T& a, T b) {
    if (a > b) {
        a = b;
        return true;
    }
    return false;
};
template<class T> inline bool chmax(T& a, T b) {
    if (a < b) {
        a = b;
        return true;
    }
    return false;
};

// intやllをstringに変換
template<typename T> inline string toString(const T &a){
  ostringstream oss;
  oss << a;
  return oss.str();
};
// ----------------------------------------------------------------------------

int gcd(int a,int b){
  if(b==0) return a;
  return gcd(b,a%b);
}

int main(){
  int N,M,K;
  cin >> N >> M >> K;
  char c;
  cin >> c;
  vector<int> B(M);
  for(int i=0; i<M; i++){
    cin >> B[i];
  }
  vector<int> A(N);
  for(int i=0; i<N; i++){
    cin >> A[i];
  }
  if(c == '*'){
    map<int,int> mp_B,mp_A;
    for(int i=0; i<M; i++){
      mp_B[gcd(K,B[i])]++;
    }
    for(int i=0; i<N; i++){
      mp_A[gcd(K,A[i])]++;
    }
    ll ans = 0;
    for(auto b: mp_B){
      for(auto a: mp_A){
        if(((ll)b.first*(ll)a.first)%K==0){
          ans += (ll)a.second*(ll)b.second;
        }
      }
    }
    cout << ans << endl;
  }else{
    vector<int> rem_B(K,0);
    vector<int> rem_A(K,0);
    for(int i=0; i<M; i++){
      rem_B[B[i]%K]++;
    }
    for(int i=0; i<N; i++){
      rem_A[A[i]%K]++;
    }
    ll ans = 0;
    ans += (ll)rem_B[0]*(ll)rem_A[0];
    for(int i=1; i<K; i++){
      ans += (ll)rem_B[i]*(ll)rem_A[K-i];
    }
    cout << ans << endl;
  }
  return 0;
}
0