test/yosupo/Math/enum_prime_4.test.cpp

View this file on GitHub
Last update: 2024-05-24 23:35:26+07:00
Problem: https://judge.yosupo.jp/problem/enumerate_primes

Depends on

Code

#define PROBLEM "https://judge.yosupo.jp/problem/enumerate_primes"
#include "Misc/marco.hpp"
#include "Misc/debug.hpp"
const int INF=1e9;
const ll INFI=1e15;
//----------Author: Nguyen Ho Nam,UIT, Saigon-----------------
#include "NT/prime/sieve/bitset_sieve_odd.hpp"
pair<int, vector<int>> enum_prime(int n) {
    if (n <= 1) return {0, {}};
    auto primes = sieve_bitset_odd(n);
    int dem = primes.size();
    return {dem, primes};
}

int main() {
    int n, a, b; cin>>n>>a>>b;
    auto [cnt, primes] = enum_prime(n);
    cout<<cnt<<" ";
    int x = 0;
    for (int i = b; i < primes.size(); i += a) {
        x++;
    }
    cout<<x<<'\n';
    for (int i = b; i < primes.size(); i += a) {
        cout<<primes[i]<<" ";
    }
    cout<<'\n';
    return 0;
}

#line 1 "test/yosupo/Math/enum_prime_4.test.cpp"
#define PROBLEM "https://judge.yosupo.jp/problem/enumerate_primes"
#line 2 "Misc/marco.hpp"
// Judges with GCC >= 12 only needs Ofast
// #pragma GCC optimize("O3,no-stack-protector,fast-math,unroll-loops,tree-vectorize")
// MLE optimization
// #pragma GCC optimize("conserve-stack")
// Old judges
// #pragma GCC target("sse4.2,popcnt,lzcnt,abm,mmx,fma,bmi,bmi2")
// New judges. Test with assert(__builtin_cpu_supports("avx2"));
// #pragma GCC target("avx2,popcnt,lzcnt,abm,bmi,bmi2,fma,tune=native")
// Atcoder
// #pragma GCC target("avx2,popcnt,lzcnt,abm,bmi,bmi2,fma")
/*
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;
typedef tree<int,null_type,less<int>,rb_tree_tag,tree_order_statistics_node_update> ods;
- insert(x),erase(x)
- find_by_order(k): return iterator to the k-th smallest element
- order_of_key(x): the number of elements that are strictly smaller
*/
#include<bits/stdc++.h>
using namespace std;

using ld = long double;
using ll = long long;
using u32 = unsigned int;
using u64 = unsigned long long;
using i128 = __int128;
using u128 = unsigned __int128;
using f128 = __float128;
 
 
#define pii pair<int,int>
#define pll pair<ll,ll>
 
#define all(x) (x).begin(),(x).end()
#define rall(x) (x).rbegin(),(x).rend()
#define ars(x) (x),(x+n)
 
#define TIME  (1.0 * clock() / CLOCKS_PER_SEC)
 
#define For(i,a,b) for (int i=(a); i<(b); i++)
#define rep(i,a) For(i,0,a)
#define rev(i,a,b) for (int i=(a); i>(b); i--)
#define FOR(i,a,b) for (int i=(a); i<=(b); i++)
#define REP(i,a) FOR(i,1,a)
#define REV(i,a,b) for (int i=(a); i>=(b); i--)
 
#define pb push_back
#define eb emplace_back
#define mp make_pair
#define fi first
#define se second
#define FT ios_base::sync_with_stdio(false); cin.tie(nullptr);
 
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

using vi=vector<int>;
using vll = vector<ll>;
template <class T>
using vc = vector<T>;
template <class T>
using vvc = vector<vc<T>>;
template <class T>
using vvvc = vector<vvc<T>>;
template <class T>
using vvvvc = vector<vvvc<T>>;
template <class T>
using vvvvvc = vector<vvvvc<T>>;
template <class T>
using pq = priority_queue<T>;
template <class T>
using pqg = priority_queue<T, vector<T>, greater<T>>;
 
#define vv(type, name, h, ...) \
  vector<vector<type>> name(h, vector<type>(__VA_ARGS__))
#define vvv(type, name, h, w, ...)   \
  vector<vector<vector<type>>> name( \
      h, vector<vector<type>>(w, vector<type>(__VA_ARGS__)))
#define vvvv(type, name, a, b, c, ...)       \
  vector<vector<vector<vector<type>>>> name( \
      a, vector<vector<vector<type>>>(       \
             b, vector<vector<type>>(c, vector<type>(__VA_ARGS__))))
 
//template <class T>
//using ods =
//   tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
 
template <typename T> bool chkmin(T &x,T y){return x>y?x=y,1:0;}
template <typename T> bool chkmax(T &x,T y){return x<y?x=y,1:0;}
 
template<class T> using pq = priority_queue<T>;
template<class T> using pqg = priority_queue<T, vector<T>, greater<T>>;
#line 1 "Misc/debug.hpp"
void __print(int x) {cerr << x;}
void __print(long x) {cerr << x;}
void __print(long long x) {cerr << x;}
void __print(unsigned x) {cerr << x;}
void __print(unsigned long x) {cerr << x;}
void __print(unsigned long long x) {cerr << x;}
void __print(float x) {cerr << x;}
void __print(double x) {cerr << x;}
void __print(long double x) {cerr << x;}
void __print(char x) {cerr << '\'' << x << '\'';}
void __print(const char *x) {cerr << '\"' << x << '\"';}
void __print(const string &x) {cerr << '\"' << x << '\"';}
void __print(bool x) {cerr << (x ? "true" : "false");}

template<typename T, typename V>
void __print(const pair<T, V> &x) {cerr << '{'; __print(x.first); cerr << ", "; __print(x.second); cerr << '}';}
template<typename T>
void __print(const T &x) {int f = 0; cerr << '{'; for (auto &i: x) cerr << (f++ ? ", " : ""), __print(i); cerr << "}";}
template<>
void __print(const vector<bool> &x) {int f = 0; cerr << '{'; for (size_t i = 0; i < x.size(); ++i) cerr << (f++ ? ", " : ""), __print(x[i]); cerr << "}";}
void _print() {cerr << "]\n";}
template <typename T, typename... V>
void _print(T t, V... v) {__print(t); if (sizeof...(v)) cerr << ", "; _print(v...);}

void dbg_out() { cerr << endl; }
template<typename Head, typename... Tail> void dbg_out(Head H, Tail... T) { __print(H); if (sizeof...(T)) cerr << ", "; dbg_out(T...); }
#define dbg(...) cerr << "[" << #__VA_ARGS__ << "]:"; dbg_out(__VA_ARGS__);
#line 4 "test/yosupo/Math/enum_prime_4.test.cpp"
const int INF=1e9;
const ll INFI=1e15;
//----------Author: Nguyen Ho Nam,UIT, Saigon-----------------
#line 2 "NT/prime/sieve/bitset_sieve_odd.hpp"
template<class T>
std::vector<T> sieve_bitset_odd(T n) {
    const T S = sizeof(uint64_t) * 8;
    std::vector<uint64_t> bitset((n / 2 + S - 1) / S, 0xFFFFFFFFFFFFFFFFULL);
    std::vector<T> primes;

    primes.push_back(2);
    for (T i = 3; i * i <= n; i += 2) {
        if (bitset[i / 2 / S] & (1ULL << ((i / 2) % S))) {
            for (T j = i * i; j <= n; j += i * 2) {
                bitset[j / 2 / S] &= ~(1ULL << ((j / 2) % S));
            }
            primes.push_back(i);
        }
    }

    for (T i = primes.back() + 2; i <= n; i += 2) {
        if (bitset[i / 2 / S] & (1ULL << ((i / 2) % S))) {
            primes.push_back(i);
        }
    }

    return primes;
}
#line 8 "test/yosupo/Math/enum_prime_4.test.cpp"
pair<int, vector<int>> enum_prime(int n) {
    if (n <= 1) return {0, {}};
    auto primes = sieve_bitset_odd(n);
    int dem = primes.size();
    return {dem, primes};
}

int main() {
    int n, a, b; cin>>n>>a>>b;
    auto [cnt, primes] = enum_prime(n);
    cout<<cnt<<" ";
    int x = 0;
    for (int i = b; i < primes.size(); i += a) {
        x++;
    }
    cout<<x<<'\n';
    for (int i = b; i < primes.size(); i += a) {
        cout<<primes[i]<<" ";
    }
    cout<<'\n';
    return 0;
}