結果
問題 | No.1661 Sum is Prime (Hard Version) |
ユーザー | 沙耶花 |
提出日時 | 2021-08-27 22:08:56 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 37 ms / 3,000 ms |
コード長 | 2,327 bytes |
コンパイル時間 | 4,266 ms |
コンパイル使用メモリ | 269,588 KB |
実行使用メモリ | 6,820 KB |
最終ジャッジ日時 | 2024-11-21 02:39:27 |
合計ジャッジ時間 | 5,558 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,816 KB |
testcase_01 | AC | 2 ms
6,816 KB |
testcase_02 | AC | 24 ms
6,820 KB |
testcase_03 | AC | 2 ms
6,820 KB |
testcase_04 | AC | 2 ms
6,820 KB |
testcase_05 | AC | 2 ms
6,820 KB |
testcase_06 | AC | 2 ms
6,816 KB |
testcase_07 | AC | 1 ms
6,816 KB |
testcase_08 | AC | 2 ms
6,820 KB |
testcase_09 | AC | 2 ms
6,820 KB |
testcase_10 | AC | 2 ms
6,820 KB |
testcase_11 | AC | 2 ms
6,816 KB |
testcase_12 | AC | 22 ms
6,816 KB |
testcase_13 | AC | 21 ms
6,820 KB |
testcase_14 | AC | 24 ms
6,820 KB |
testcase_15 | AC | 31 ms
6,820 KB |
testcase_16 | AC | 26 ms
6,816 KB |
testcase_17 | AC | 24 ms
6,816 KB |
testcase_18 | AC | 11 ms
6,816 KB |
testcase_19 | AC | 18 ms
6,820 KB |
testcase_20 | AC | 12 ms
6,820 KB |
testcase_21 | AC | 20 ms
6,820 KB |
testcase_22 | AC | 16 ms
6,816 KB |
testcase_23 | AC | 37 ms
6,820 KB |
ソースコード
#include <stdio.h> #include <bits/stdc++.h> #include <atcoder/all> using namespace atcoder; using mint = modint1000000007; using namespace std; #define rep(i,n) for (int i = 0; i < (n); ++i) #define Inf 1000000001 using i64 = long long; int isqrt(i64 n) { return sqrtl(n); } __attribute__((target("avx2"), optimize("O3", "unroll-loops"))) i64 prime_pi(const i64 N) { if (N <= 1) return 0; if (N == 2) return 1; const int v = isqrt(N); int s = (v + 1) / 2; vector<int> smalls(s); for (int i = 1; i < s; ++i) smalls[i] = i; vector<int> roughs(s); for (int i = 0; i < s; ++i) roughs[i] = 2 * i + 1; vector<i64> larges(s); for (int i = 0; i < s; ++i) larges[i] = (N / (2 * i + 1) - 1) / 2; vector<bool> skip(v + 1); const auto divide = [] (i64 n, i64 d) -> int { return double(n) / d; }; const auto half = [] (int n) -> int { return (n - 1) >> 1; }; int pc = 0; for (int p = 3; p <= v; p += 2) if (!skip[p]) { int q = p * p; if (i64(q) * q > N) break; skip[p] = true; for (int i = q; i <= v; i += 2 * p) skip[i] = true; int ns = 0; for (int k = 0; k < s; ++k) { int i = roughs[k]; if (skip[i]) continue; i64 d = i64(i) * p; larges[ns] = larges[k] - (d <= v ? larges[smalls[d >> 1] - pc] : smalls[half(divide(N, d))]) + pc; roughs[ns++] = i; } s = ns; for (int i = half(v), j = ((v / p) - 1) | 1; j >= p; j -= 2) { int c = smalls[j >> 1] - pc; for (int e = (j * p) >> 1; i >= e; --i) smalls[i] -= c; } ++pc; } larges[0] += i64(s + 2 * (pc - 1)) * (s - 1) / 2; for (int k = 1; k < s; ++k) larges[0] -= larges[k]; for (int l = 1; l < s; ++l) { int q = roughs[l]; i64 M = N / q; int e = smalls[half(M / q)] - pc; if (e < l + 1) break; i64 t = 0; for (int k = l + 1; k <= e; ++k) t += smalls[half(divide(M, roughs[k]))]; larges[0] += t - i64(e - l) * (pc + l - 1); } return larges[0] + 1; } int main(){ long long L,R; cin>>L>>R; long long ans = 0; if(R-L<=5){ for(long long i=L;i<=R;i++){ if(prime_pi(i)-prime_pi(i-1)==1)ans++; } for(long long i=L;i<R;i++){ if(prime_pi(i*2+1)-prime_pi(i*2)==1)ans++; } } else{ ans += prime_pi(R); ans -= prime_pi(L-1); ans += prime_pi(R*2-1); ans -= prime_pi(L*2); } cout<<ans<<endl; return 0; }