結果

問題 No.990 N×Mマス計算（Kの倍数）
ユーザー kappybar
提出日時 2020-04-08 11:08:06
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 209 ms / 2,000 ms
コード長 1,176 bytes
コンパイル時間 1,630 ms
コンパイル使用メモリ 177,676 KB
実行使用メモリ 16,512 KB
最終ジャッジ日時 2024-07-18 07:47:32
合計ジャッジ時間 3,925 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define rep(i,n) for(int i=0;i<n;i++)
using namespace std;
using ll =  long long ;
using P = pair<int,int> ;
const int INF = 1e9;
const int MOD = 1000000007;

ll gcd(ll i,ll j){
        if(j == 0) return i;
        return gcd(j,i%j);
}

int main(){
    ll n,m,k;
    cin >> n >> m >> k;
    char op;
    cin >> op;
    ll ans = 0;
    if(op == '+'){
        map<ll,ll> b;
        vector<ll> a(n);
        rep(i,m){
            ll p;
            cin >> p;
            p %= k;
            b[p] ++;
        }
        rep(i,n) cin >> a[i];
        rep(i,n){
            a[i] %= k;
            if(a[i] == 0) ans += b[0];
            else ans += b[k-a[i]];
        }
    }else{
        map<ll,ll> a,b;
        rep(i,m){
            ll p;
            cin >> p;
            p = gcd(p,k);
            b[p] ++;
        }rep(i,n){
            ll p;
            cin >> p;
            p = gcd(p,k);
            a[p] ++;
        }
        for(auto p:a){
            for(auto q:b){
                if((p.first * q.first)%k == 0){
                    ans += p.second * q.second;
                }
            }
        }
    }
    cout << ans << endl;
    return 0;
}
0