結果

問題 No.990 N×Mマス計算（Kの倍数）
コンテスト
ユーザー zeronosu77108
提出日時 2020-02-14 22:30:49
言語 C++17(clang)
(17.0.6 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 879 bytes
コンパイル時間 599 ms
コンパイル使用メモリ 120,960 KB
実行使用メモリ 817,792 KB
最終ジャッジ日時 2024-11-30 16:23:31
合計ジャッジ時間 5,281 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 11 WA * 3 RE * 1 MLE * 4
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <numeric>

using namespace std;

using int64 = long long;

int main() {
    cin.tie(nullptr); ios::sync_with_stdio(false);
    int n,m,k;
    cin >> n >> m >> k;
    char op; cin >> op; bool plus = op == '+';

    vector<int> bk(k,0);
    vector<int> ak(k,0);

    int i,j;
    int a,b;
    for (i=0; i<m; i++) {
        cin >> b;
        bk[b % k]++;
    }

    for (i=0; i<n; i++) {
        cin >> a;
        ak[a % k]++;
    }


    int64 ans = 0;
    if (plus) {
        for (i=0; i<k; i++) {
            ans += ak[i] * bk[(k-i)%k];
        }
    } else {
        ans += (int64)ak[0] * m;
        ans += (int64)bk[0] * n;
        for (i=2; i<k; i++) {
            int tmp = k / gcd(k, i);
            for (j=tmp; j<k; j+=tmp) {
                ans += (int64)ak[i]*bk[j];
            }
        }
    }

    cout << ans << endl;
}
0