結果

問題 No.990 N×Mマス計算（Kの倍数）
コンテスト
ユーザー face4
提出日時 2020-02-18 19:53:59
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,456 ms / 2,000 ms
コード長 1,190 bytes
コンパイル時間 740 ms
コンパイル使用メモリ 89,240 KB
実行使用メモリ 14,208 KB
最終ジャッジ日時 2024-11-16 02:55:09
合計ジャッジ時間 5,709 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<vector>
#include<algorithm>
#include<map>
using namespace std;
typedef long long ll;

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

int main(){
    int n, m, k;
    char op;
    cin >> n >> m >> k >> op;
    vector<ll> b(m), a(n);
    for(int i = 0; i < m; i++)  cin >> b[i];
    for(int i = 0; i < n; i++)  cin >> a[i];
    ll ret = 0;
    if(op == '+'){
        map<int,int> cnt;
        for(int i = 0; i < m; i++)  cnt[b[i]%k]++;
        for(int i = 0; i < n; i++)  ret += cnt[(k-a[i]%k)%k];
    }else if(op == '*'){
        vector<int> fact;
        for(ll i = 1; i*i <= k; i++){
            if(k%i) continue;
            fact.push_back(i);
            if(i != k/i)    fact.push_back(k/i);
        }
        vector<int> v;
        for(int j : fact){
            int tmp = 0;
            for(int i = 0; i < m; i++)  if(b[i]%j==0)   tmp++;
            v.push_back(tmp);
        }
        map<int,int> cnt;
        for(int i = 0; i < fact.size(); i++)    cnt[fact[i]] = v[i];
        for(int i = 0; i < n; i++){
            ll g = gcd(k, a[i]);
            g = k/g;
            ret += cnt[g];
        }
    }
    cout << ret << endl;
    return 0;
}
0