結果
問題 |
No.990 N×Mマス計算(Kの倍数)
|
ユーザー |
|
提出日時 | 2023-07-27 02:59:05 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 67 ms / 2,000 ms |
コード長 | 1,500 bytes |
コンパイル時間 | 1,831 ms |
コンパイル使用メモリ | 175,352 KB |
実行使用メモリ | 6,824 KB |
最終ジャッジ日時 | 2024-10-03 16:28:17 |
合計ジャッジ時間 | 3,253 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 19 |
ソースコード
#include <bits/stdc++.h> using namespace std; using ll = long long; int main(){ ios::sync_with_stdio(false); cin.tie(0); int n, m, k; char c; cin >> n >> m >> k >> c; vector<int> a(n), b(m); for(auto &&v : b) cin >> v; for(auto &&v : a) cin >> v; if(c == '+'){ ll ans = 0; for(int i = 0; i < n; i++) a[i] %= k; sort(a.begin(), a.end()); for(int i = 0; i < m; i++){ b[i] %= k; auto pa = equal_range(a.begin(), a.end(), k - b[i] == k ? 0 : k - b[i]); ans += pa.second - pa.first; } cout << ans << '\n'; }else{ vector<int> divs; for(int i = 1; i * i <= k; i++){ if(k % i == 0){ divs.emplace_back(i); if(k != i * i) divs.emplace_back(k / i); } } sort(divs.begin(), divs.end()); int cn = divs.size(); vector<int> cnt(cn); for(int i = 0; i < n; i++){ int gcdv = __gcd(a[i], k); cnt[lower_bound(divs.begin(), divs.end(), gcdv) - divs.begin()]++; } for(int i = 1; i < cn; i++){ for(int j = 0; j < i; j++){ if(divs[i] % divs[j] == 0) cnt[j] += cnt[i]; } } ll ans = 0; for(int i = 0; i < m; i++){ int v = k / __gcd(b[i], k); ans += cnt[lower_bound(divs.begin(), divs.end(), v) - divs.begin()]; } cout << ans << '\n'; } }