結果
問題 | No.990 N×Mマス計算(Kの倍数) |
ユーザー |
![]() |
提出日時 | 2020-02-14 21:51:17 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 1,012 ms / 2,000 ms |
コード長 | 2,474 bytes |
コンパイル時間 | 2,554 ms |
コンパイル使用メモリ | 190,864 KB |
実行使用メモリ | 14,080 KB |
最終ジャッジ日時 | 2024-11-16 00:34:51 |
合計ジャッジ時間 | 5,554 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 19 |
ソースコード
#include<bits/stdc++.h> using namespace std; typedef long long ll; typedef pair<int,int> P; #define p_ary(ary,a,b) do { cout << "["; for (int count = (a);count < (b);++count) cout << ary[count] << ((b)-1 == count ? "" : ", "); cout << "]\n"; } while(0) #define p_map(map,it) do {cout << "{";for (auto (it) = map.begin();;++(it)) {if ((it) == map.end()) {cout << "}\n";break;}else cout << "" << (it)->first << "=>" << (it)->second << ", ";}}while(0) ll gcd(ll a,ll b) { while (b) { a %= b; swap(a,b); } return a; } int const MAX_N = 100000; vector<ll> prime; vector<bool> is_prime(MAX_N,true); void Eratosthenes() { is_prime[0] = is_prime[1] = false; for (ll i = 2;i*i < MAX_N;++i) if (is_prime[i]) for (ll j = 2*i;j < MAX_N;j += i) is_prime[j] = false; for (ll i = 0;i < MAX_N;++i) if (is_prime[i]) prime.push_back(i); } vector<P> prime_factorization(ll a) { vector<P> prime_factor; if (is_prime[0]) Eratosthenes(); for (ll& i : prime) { if (i*i > a) break; if (a%i == 0) { P p = P(i,0); while (a%i == 0) { a /= i; p.second++; } prime_factor.push_back(p); } } if (a != 1) prime_factor.push_back(P(a,1)); return prime_factor; } void rec(vector<ll>& div,vector<P>& fact,int i,ll d) { if (i == fact.size()) { div.push_back(d); return; } for (int j = 0;j <= fact[i].second;++j) { rec(div,fact,i+1,d); d *= fact[i].first; } } // unsorted vector<ll> divisor(ll a) { vector<ll> div; vector<P> prime_factor = prime_factorization(a); rec(div,prime_factor,0,1); return div; } int main() { int n,m,k; char c; cin >> n >> m >> k >> c; vector<ll> a(n),b(m); for (int i = 0;i < m;++i) cin >> b[i]; for (int i = 0;i < n;++i) cin >> a[i]; if (c == '+') { ll ans = 0; map<int,int> mp; for (int i = 0;i < m;++i) mp[b[i]%k]++; for (int i = 0;i < n;++i) ans += mp[(k-a[i]%k)%k]; cout << ans << endl; } else { ll ans = 0; map<ll,ll> mp; for (int i = 0;i < m;++i) { b[i] = gcd(b[i],k); for (ll j : divisor(gcd(b[i],k))) mp[j]++; } for (int i = 0;i < n;++i) { if (a[i]%k == 0) ans += m; else { ans += mp[k/gcd(a[i],k)]; } } cout << ans << endl; } }