結果

問題 No.990 N×Mマス計算（Kの倍数）
コンテスト
ユーザー tamaron
提出日時 2020-02-14 22:53:07
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
MLE  
実行時間 -
コード長 2,725 bytes
コンパイル時間 2,008 ms
コンパイル使用メモリ 183,440 KB
実行使用メモリ 703,140 KB
最終ジャッジ日時 2024-11-16 01:15:34
合計ジャッジ時間 20,911 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 MLE * 1
other AC * 10 WA * 2 TLE * 5 MLE * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define rep(i,n) for(int i=0;i<n;++i)
#define rep1(i,n) for(int i=1;i<=n;++i)
#define eachdo(e,array) for(const auto& e : array)
#define lower_index(v,a) (int)distance(v.begin(), lower_bound((v).begin(),(v).end(),a))
template<class T> bool chmin(T &a, T b){if(a>b){a=b;return true;}return false;}
typedef long long ll;
const ll INF = 1e18;
const ll MOD = 1e9+7;
const ll MAX = 400005;
long long fac[MAX], finv[MAX], inv[MAX];


ll modpow(ll a, ll n, ll mod=MOD) {
    ll res = 1;
    while (n > 0) {
        if (n & 1) res = res * a % mod;
        a = a * a % mod;
        n >>= 1;
    }
    return res;
}
ll modinv(ll a,ll m=MOD) {
    ll b=m,u=1,v=0;
    while(b){
        ll t=a/b;
        a-=t*b;
        swap(a,b);
        u-=t*v;
        swap(u,v);
    }
    u%=m;
    if(u<0) u+=m;
    return u;
}

void COMinit(){
    fac[0]=fac[1]=1;
    finv[0]=finv[1]=1;
    inv[1]=1;
    for (ll i=2;i<MAX;i++){
        fac[i]=fac[i-1]*i%MOD;
        inv[i] =MOD-inv[MOD%i]*(MOD/i)%MOD;
        finv[i]=finv[i-1]*inv[i]%MOD;
    }
}

ll COM(ll n, ll k){
    if (n<k)
        return 0;
    if (n<0 || k<0)
        return 0;
    return fac[n]*(finv[k]*finv[n - k]%MOD)%MOD;
}

set<ll> division(ll n){
    set<ll> S;
    ll num = 1;
    ll root = (ll)(sqrt(n));
    while(num<=root){
        if (n%num == 0){
            S.insert(num);
            S.insert(n/num);
        }
        num++;
    }
    return S;
}

int main(){ 
    ll H,W,K;cin >> H >> W >> K;
    set<ll> set = division(K);
    ll root = (ll)(sqrt(K));
    string op;cin >> op;
    vector<ll> h(H),w(W);
    rep(i,W) cin >> w[i];
    rep(i,H) cin >> h[i];
    //sort(h.begin(),h.end());
    //sort(w.begin(),w.end());

    ll res = 0;
    if(op == "+"){
        map<ll,ll> hash,hasw;
        rep(i,H){
            hash[h[i] % K] += 1;
        }
        rep(i,W){
            hasw[w[i] % K] += 1;
        }
        rep(i,K){
            res += hash[i % K] * hasw[(K-i) % K];
        }
    }else{
        map<ll,ll> hash,hasw;
        rep1(i,root){
            if(set.count(i)){
                rep(j,H){
                    if(h[j] % i == 0) hash[i] += 1;
                    if(h[j] % (K / i) == 0 && i != (K / i)) hash[K / i] += 1;
                }
                rep(j,W){
                    if(w[j] % i == 0) hasw[i] += 1;
                    if(w[j] % (K / i) == 0 && i != (K / i)) hasw[K / i] += 1;
                }
            }
        }
        for(auto p : hash){
            auto key = p.first;
            auto value = p.second;
            // cout << key << " "<< value << " " <<  K/key << " " << hasw[K /key] << endl;
            res += value * hasw[K / key];
        }
    }
    cout << res << endl;

}
0