結果

問題 No.990 N×Mマス計算（Kの倍数）
ユーザー pitsu_kyo_propitsu_kyo_pro
提出日時 2020-02-14 22:57:24
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 2,816 bytes
コンパイル時間 955 ms
コンパイル使用メモリ 116,560 KB
実行使用メモリ 813,936 KB
最終ジャッジ日時 2024-04-27 20:05:10
合計ジャッジ時間 3,001 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 WA -
testcase_03 AC 2 ms
6,940 KB
testcase_04 AC 1 ms
6,940 KB
testcase_05 AC 2 ms
6,940 KB
testcase_06 WA -
testcase_07 WA -
testcase_08 AC 2 ms
6,940 KB
testcase_09 AC 4 ms
6,940 KB
testcase_10 AC 82 ms
135,776 KB
testcase_11 WA -
testcase_12 WA -
testcase_13 AC 27 ms
6,944 KB
testcase_14 AC 39 ms
6,940 KB
testcase_15 AC 26 ms
6,940 KB
testcase_16 MLE -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
権限があれば一括ダウンロードができます

ソースコード

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();
};
// ----------------------------------------------------------------------------
// beetさんの素因数分解
template<typename T>
map<T, ll> factorize(T x){
  map<T, ll> res;
  for(ll i=2;i*i<=x;i++){
    while(x%i==0){
      x/=i;
      res[i]++;
    }
  }
  if(x!=1) res[x]++;
  return res;
}
// 使用例
// vector<int> cnt(111,0);
// for(int i=1; i<=N; i++){
//   for(auto p: factorize(i)){
//     cnt[p.first] += p.second;
//   }
// }

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_K;
    mp_K = factorize(K);
    map<pair<ll,ll>,ll> mp;
    for(int i=0; i<M; i++){
      for(auto p: mp_K){
        if(B[i]%p.first == 0){
          int cnt=0;
          ll temp = B[i];
          while(temp%p.first==0){
            cnt++;
            temp /= p.first;
            mp[make_pair(p.first,cnt)]++;
          }
        }
        mp[make_pair(p.first,0)]++;
      }
    }
      ll ans = 0;
      for(int i=0; i<N; i++){
        ll res = 1e9;
        for(auto p: mp_K){
          ll temp = A[i];
          ll cnt=0;
          while(temp%p.first == 0){
            cnt++;
            temp /= p.first;
          }
          chmin(res,mp[make_pair(p.first,(ll)(max(0LL,p.second-cnt)))]);
        }
        ans += res;
      }
    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