結果

問題 No.990 N×Mマス計算（Kの倍数）
ユーザー seriru13
提出日時 2020-02-16 10:42:27
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 237 ms / 2,000 ms
コード長 1,428 bytes
コンパイル時間 2,102 ms
コンパイル使用メモリ 208,420 KB
最終ジャッジ日時 2025-01-09 00:53:49
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define inf 10e17
#define rep(i,n) for(long long i=0; i<n; i++)
#define repr(i,n,m) for(long long i=m; i<n; i++)
#define mod 1000000007
#define sorti(x) sort(x.begin(), x.end())
#define sortd(x) sort(x.begin(), x.end(), std::greater<long long>())
#define debug(x) std::cerr << (x) << std::endl;
#define roll(x) for (auto&& itr : x) { cerr << (itr) << " "; }

template <class T> inline void chmax(T &ans, T t) { if (t > ans) ans = t;}
template <class T> inline void chmin(T &ans, T t) { if (t < ans) ans = t;}

// 最大公約数を求める(ユークリッドの互除法).
long long gcd (long long x, long long y) {
  if (y > x) swap(x,y);
  if (y == 0) return x;
  return gcd(x%y,y);
}

int main() {
  int n,m,k;
  cin >> n >> m >> k;
  char op;
  cin >> op;
  vector<int> a(n), b(m);
  map<int, int> mp;
  rep(i, m) {
    cin >> b[i];
    mp[b[i] % k] += 1;
  }
  rep(i, n) {
    cin >> a[i];
  }

  ll ans = 0;
  if (op == '+') {
    for (auto itr : a) {
      ans += mp[k - 1 - (itr - 1) % k];
    }
  } else {
    map<ll, ll> am, bm;
    for (auto itr : a) {
      am[gcd(itr, k)] += 1;
    }

    for (auto itr : b) {
      bm[gcd(itr, k)] += 1;
    }

    for (auto itr : am) {
      for (auto itr2 : bm) {
        if ((itr.first * itr2.first) % k == 0) {
          ans += itr.second * itr2.second;
        }
      }
    }
  }

  cout << ans << endl;
}
0