結果

問題 No.990 N×Mマス計算（Kの倍数）
ユーザー tactac
提出日時 2020-02-14 22:29:07
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 3,302 bytes
コンパイル時間 2,001 ms
コンパイル使用メモリ 186,760 KB
実行使用メモリ 13,440 KB
最終ジャッジ日時 2024-04-27 19:50:04
合計ジャッジ時間 7,802 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,948 KB
testcase_02 AC 2 ms
6,944 KB
testcase_03 AC 2 ms
6,940 KB
testcase_04 AC 2 ms
6,944 KB
testcase_05 AC 2 ms
6,940 KB
testcase_06 AC 2 ms
6,940 KB
testcase_07 AC 2 ms
6,940 KB
testcase_08 AC 2 ms
6,944 KB
testcase_09 AC 2 ms
6,944 KB
testcase_10 AC 38 ms
6,940 KB
testcase_11 AC 19 ms
6,944 KB
testcase_12 TLE -
testcase_13 AC 15 ms
6,940 KB
testcase_14 WA -
testcase_15 WA -
testcase_16 AC 45 ms
7,168 KB
testcase_17 WA -
testcase_18 TLE -
testcase_19 AC 51 ms
6,940 KB
testcase_20 AC 157 ms
13,440 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define F first
#define S second
#define pii pair<int, int>
#define eb emplace_back
#define all(v) v.begin(), v.end()
#define rep(i, n) for (int i = 0; i < (n); ++i)
#define rep3(i, l, n) for (int i = l; i < (n); ++i)
#define sz(v) (int)v.size()
const int inf = 1e9 + 7;
const ll INF = 1e18;
#define abs(x) (x >= 0 ? x : -(x))
#define lb(v, x) (int)(lower_bound(all(v), x) - v.begin())
#define ub(v, x) (int)(upper_bound(all(v), x) - v.begin())
template<typename T1, typename T2> inline bool chmin(T1 &a, T2 b) { if (a > b) { a = b; return 1; } return 0; }
template<typename T1, typename T2> inline bool chmax(T1 &a, T2 b) { if (a < b) { a = b; return 1; } return 0; }
template<typename T> T gcd(T a, T b) { if (b == 0) return a; return gcd(b, a % b); }
template<typename T> T lcm(T a, T b) { return a / gcd(a, b) * b; }
template<typename T> T pow(T a, int b) { return b ? pow(a * a, b / 2) * (b % 2 ? a : 1) : 1; }
const int mod = 1000000007;
ll modpow(ll a, int b) { return b ? modpow(a * a % mod, b / 2) * (b % 2 ? a : 1) % mod : 1; }
template<class T> ostream& operator<<(ostream& os, const vector<T>& vec) { for (auto &vi: vec) os << vi << " "; return os; }
template<class T, class U> ostream& operator<<(ostream& os, const pair<T, U>& p) { os << p.F << " " << p.S; return os; }
template<class T> inline void add(T &a, int b) { a += b; if (a >= mod) a -= mod; }



void solve();

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);

    int T;
    // cin >> T;
    T = 1;

    while (T--) {
        solve();
    }
}

template<typename T>
map<T, int> prime_factorize(T n) {
    T tmp = n;
    map<T, int> mp;
    for (T i = 2; i * i <= n; ++i) {
        ll cnt = 0;
        while (tmp % i == 0) {
            tmp /= i;
            cnt++;
        }
        if (cnt) mp[i] += cnt;
    }
    if (tmp != 1) mp[tmp]++;
    return mp;
}

set<int> all_divisor_catch(int n) {
    set<int> s;
    for (int i = 1; i * i <= n; ++i) {
        if (n % i == 0) {
            s.insert(i);
            s.insert(n / i);
        }
    }
    return s;
}

random_device rnd;
mt19937 mt(rnd());
uniform_int_distribution<> rand100(1, 1000000000);
#define rm rand100(mt)

void solve() {
  int n, m, k;
  cin >> n >> m >> k;
  char op;
  cin >> op;
  vector<int> b(m), a(n);
  rep(i, m) cin >> b[i];
  rep(i, n) cin >> a[i];

  if (op == '+') {
    ll ans = 0;
    map<int, int> mp;
    rep(i, m) mp[b[i] % k]++;
    rep(i, n) ans += mp[k - a[i] % k];
    cout << ans << endl;
  } else {
    ll ans = 0;
    // 計算量的に素因数分解不能ということが判明
    // O(mlogbi)

    // 必要な素因数だけなら調査できる
    auto mpk = prime_factorize(k);
    map<int, int> mpv;

    rep(i, n) {
      map<int, int> mp;
      for (auto e : mpk) {
        int cnt = 0;
        while (a[i] % e.F == 0) {
          a[i] /= e.F;
          cnt++;
        }
        if (cnt) mp[e.F] = min(cnt, e.S);
      }
      int x = 1;
      for (auto e : mp) x *= pow(e.F, e.S);
      auto s = all_divisor_catch(x);
      for (auto e : s) mpv[e]++;
    }
    // for (auto e : mpv) cout << e.F << " " << e.S << endl;
    rep(i, m) {
      int g = gcd(k, b[i]);
      ans += mpv[k / g];
    }
    cout << ans << endl;
  }
}
0