結果

問題 No.990 N×Mマス計算（Kの倍数）
ユーザー leaf_1415
提出日時 2020-02-14 21:46:42
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 1,692 ms / 2,000 ms
コード長 1,379 bytes
コンパイル時間 886 ms
コンパイル使用メモリ 79,996 KB
実行使用メモリ 23,680 KB
最終ジャッジ日時 2024-11-16 00:31:06
合計ジャッジ時間 6,512 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <map>
#include <vector>
#include <algorithm>
#define llint long long

using namespace std;

llint h, w, k;
char op;
llint a[100005], b[100005];
vector<llint> vec;
llint num[2005];
map<llint, llint> mp, mpA, mpB;

llint gcd(llint a,llint b)
{
  if(b == 0) return a;
  return gcd(b, a%b);
}

int main(void)
{
  ios::sync_with_stdio(0);
  cin.tie(0);

  cin >> h >> w >> k;
  cin >> op;
  for(int i = 1; i <= w; i++) cin >> a[i];
  for(int i = 1; i <= h; i++) cin >> b[i];

  for(int i = 1; i <= w; i++) mpA[a[i]%k]++;
  for(int i = 1; i <= h; i++) mpB[b[i]%k]++;

  if(op == '+'){
    llint ans = 0;
    for(auto it = mpA.begin(); it != mpA.end(); it++){
      ans += it->second * mpB[(k-it->first) % k];
    }
    cout << ans << endl;
    return 0;
  }
  else{

    for(int i = 1; i*i <= k; i++){
      if(k % i == 0){
        vec.push_back(i);
        vec.push_back(k/i);
      }
    }
    sort(vec.begin(), vec.end());
    vec.erase(unique(vec.begin(), vec.end()), vec.end());

    llint m = vec.size();
    for(int i = 0; i < m; i++) mp[vec[i]] = i;
    for(int i = 1; i <= w; i++){
      for(int j = 0; j < m; j++){
        if(a[i] % vec[j] == 0) num[j]++;
      }
    }

    llint ans = 0;
    for(int i = 1; i <= h; i++){
      llint g = gcd(k, b[i]);
      ans += num[mp[k / g]];
    }
    cout << ans << endl;
    return 0;
  }

  return 0;
}
0