結果

問題 No.990 N×Mマス計算（Kの倍数）
コンテスト
ユーザー auaua
提出日時 2020-02-14 22:19:48
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,627 bytes
コンパイル時間 2,235 ms
コンパイル使用メモリ 183,696 KB
実行使用メモリ 17,280 KB
最終ジャッジ日時 2024-11-16 00:51:44
合計ジャッジ時間 8,098 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 16 WA * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
#include<unordered_set>
#include<unordered_map>
using namespace std;
#define REP(i,m,n) for(int i=(m);i<(n);i++)
#define rep(i,n) REP(i,0,n)
#define pb push_back
#define all(a) a.begin(),a.end()
#define rall(c) (c).rbegin(),(c).rend()
#define mp make_pair
#define double long double
typedef long long ll;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
typedef vector<ll> vec;
typedef vector<vec> mat;
const ll inf=1e9+7;
const ll mod=1e9+7;
//gcd
ll gcd(ll a,ll b){
    if(min(a,b)==0)return max(a,b);
    if(max(a,b)%min(a,b)==0)return min(a,b);
    return gcd(min(a,b),max(a,b)%min(a,b));
}
int main(){
    ll n,m,k;cin>>n>>m>>k;
    vector<ll>a(n),b(m);
    char c;cin>>c;
    bool f=c=='+';
    rep(i,m)cin>>b[i];
    rep(i,n)cin>>a[i];
    if(f){
        map<ll,ll>ma;
        rep(i,m){
            ll s=b[i]%k;
            if(ma.find(s)!=ma.end())ma[s]++;
            else ma[s]=1;
        }
        ll ans=0;
        rep(i,n){
            ll s=a[i]%k;
            ans=ans+ma[k-s];
        }
        cout<<ans<<endl;
    }else{
        vector<ll>c(0);
        REP(i,1,sqrt(k)+1){
            if(k%i==0){
                c.pb(i);
                c.pb(k/i);
            }
        }
        sort(all(c));
        map<ll,ll>ma;
        vector<ll>d(c.size());
        rep(i,c.size()){
            rep(j,m){
                if(b[j]%c[i]==0)d[i]++;
            }
        }
        rep(i,c.size()){
            if(ma.find(c[i])==ma.end())ma[c[i]]=d[i];
        }
        ll ans=0;
        rep(i,n){
            ll s=gcd(k,a[i]);
            ans=ans+ma[k/s];
        }
        cout<<ans<<endl;
    }
}
0