結果
問題 |
No.732 3PrimeCounting
|
ユーザー |
![]() |
提出日時 | 2023-09-22 00:08:38 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 439 ms / 3,000 ms |
コード長 | 1,975 bytes |
コンパイル時間 | 6,071 ms |
コンパイル使用メモリ | 320,368 KB |
実行使用メモリ | 18,168 KB |
最終ジャッジ日時 | 2024-07-07 17:03:52 |
合計ジャッジ時間 | 12,665 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 89 |
ソースコード
#include <atcoder/all> #include <bits/stdc++.h> using ll = long long; #define rep(i, n) for(ll i = 0; i < (ll)(n); i++) #define REP(i, m, n) for(ll i = (ll)(m); i < (ll)(n); i++) using namespace std; using namespace atcoder; using mint = modint998244353; tuple<vector<ll>, vector<ll>> solve(int l, int r, vector<int> &f) { if(r - l == 1) { return {{0}, {0}}; } int m = (l + r) / 2; auto [l2, l3] = solve(l, m, f); auto [r2, r3] = solve(m, r, f); vector<ll> a2(2 * (r - l) - 1), a3(3 * (r - l) - 2); vector<ll> fl(m - l), fr(r - m); rep(i, m - l) { fl[i] = f[l + i]; } rep(i, r - m) { fr[i] = f[m + i]; } { auto n2 = convolution(fl, fr); rep(i, n2.size()) { a2[m - l + i] += n2[i]; } rep(i, l2.size()) { a2[i] += l2[i]; } rep(i, r2.size()) { a2[2 * (m - l) + i] += r2[i]; } } { auto n2_1 = convolution(l2, fr); auto n1_2 = convolution(fl, r2); rep(i, n2_1.size()) { a3[m - l + i] += n2_1[i]; } rep(i, n1_2.size()) { a3[2 * (m - l) + i] += n1_2[i]; } rep(i, l3.size()) { a3[i] += l3[i]; } rep(i, r3.size()) { a3[3 * (m - l) + i] += r3[i]; } } return {a2, a3}; } int main() { ios::sync_with_stdio(false); cin.tie(0); cout.tie(0); int n; cin >> n; vector<int> is_prime(3 * n + 1, 1); is_prime[0] = is_prime[1] = 0; for(int i = 2; i <= 3 * n; i++) { if(!is_prime[i]) { continue; } for(int j = 2 * i; j <= 3 * n; j += i) { is_prime[j] = 0; } } vector<int> f(n + 1); rep(i, n + 1) f[i] = is_prime[i]; auto ret = get<1>(solve(0, n + 1, f)); ll ans = 0; rep(i, 3 * n + 1) { ans += is_prime[i] ? ret[i] : 0; } cout << ans << endl; return 0; }