結果

問題 No.990 N×Mマス計算（Kの倍数）
ユーザー pitsu_kyo_pro
提出日時 2020-02-15 11:13:37
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
MLE  
実行時間 -
コード長 2,074 bytes
コンパイル時間 1,322 ms
コンパイル使用メモリ 108,916 KB
実行使用メモリ 814,100 KB
最終ジャッジ日時 2024-11-16 02:09:54
合計ジャッジ時間 4,795 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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();
};
// ----------------------------------------------------------------------------

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

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