結果

問題 No.990 N×Mマス計算（Kの倍数）
ユーザー knshnb
提出日時 2020-02-20 21:05:16
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
RE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,466 bytes
コンパイル時間 2,195 ms
コンパイル使用メモリ 204,512 KB
最終ジャッジ日時 2025-01-09 01:10:53
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 18 RE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>  // clang-format off
using namespace std;
using Int = int;
#define REP2(i, n) for (Int i = 0, max_i = (n); i < max_i; i++)
#define REP3(i, a, b) for (Int i = (a), max_i = (b); i < max_i; i++)
#define OVERLOAD2(_1, _2, _3, name, ...) name
#define REP(...) OVERLOAD2(__VA_ARGS__, REP3, REP2)(__VA_ARGS__)
struct SetupIO { SetupIO() { cin.tie(nullptr), ios::sync_with_stdio(false), cout << fixed << setprecision(13); } } setup_io;
#ifndef _MY_DEBUG
#define dump(...)
#endif  // clang-format on

/**
 *    author:  knshnb
 *    created: Fri Feb 14 21:57:55 JST 2020
 **/

signed main() {
    Int n, m, K;
    cin >> n >> m >> K;
    char op;
    cin >> op;
    vector<Int> b(m);
    REP(j, m) cin >> b[j];
    vector<Int> a(n);
    REP(i, n) cin >> a[i];
    sort(a.begin(), a.end());
    sort(b.begin(), b.end());
    long long ans = 0;
    if (op == '+') {
        vector<Int> num(K);
        REP(j, m) {
            num[b[j] % K]++;
        }
        REP(i, n) {
            Int tmp = (K - a[i] % K) % K;
            ans += num[tmp];
        }
    } else {
        unordered_map<Int, Int> c, d;
        REP(i, n) {
            c[__gcd(K, a[i])]++;
        }
        REP(j, m) {
            d[__gcd(K, b[j])]++;
        }
        for (auto& p1 : c) {
            for (auto& p2 : d) {
                if ((long long)p1.first * p2.first % K == 0) ans += (long long)p1.second * p2.second;
            }
        }
    }
    cout << ans << endl;
}
0