結果

問題 No.843 Triple Primes
ユーザー KoDKoD
提出日時 2021-03-26 10:24:47
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 13 ms / 2,000 ms
コード長 3,645 bytes
コンパイル時間 678 ms
コンパイル使用メモリ 79,820 KB
実行使用メモリ 10,212 KB
最終ジャッジ日時 2023-08-18 15:35:35
合計ジャッジ時間 2,905 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,380 KB
testcase_01 AC 13 ms
10,212 KB
testcase_02 AC 2 ms
4,380 KB
testcase_03 AC 2 ms
4,376 KB
testcase_04 AC 2 ms
4,376 KB
testcase_05 AC 2 ms
4,380 KB
testcase_06 AC 2 ms
4,380 KB
testcase_07 AC 13 ms
10,052 KB
testcase_08 AC 13 ms
10,212 KB
testcase_09 AC 13 ms
10,152 KB
testcase_10 AC 12 ms
10,004 KB
testcase_11 AC 12 ms
10,204 KB
testcase_12 AC 13 ms
10,144 KB
testcase_13 AC 13 ms
9,996 KB
testcase_14 AC 13 ms
10,060 KB
testcase_15 AC 12 ms
10,056 KB
testcase_16 AC 13 ms
10,060 KB
testcase_17 AC 1 ms
4,376 KB
testcase_18 AC 1 ms
4,376 KB
testcase_19 AC 2 ms
4,376 KB
testcase_20 AC 4 ms
4,992 KB
testcase_21 AC 5 ms
4,792 KB
testcase_22 AC 13 ms
9,996 KB
testcase_23 AC 12 ms
10,212 KB
testcase_24 AC 8 ms
6,464 KB
testcase_25 AC 5 ms
4,788 KB
testcase_26 AC 12 ms
10,200 KB
testcase_27 AC 2 ms
4,376 KB
testcase_28 AC 13 ms
10,076 KB
testcase_29 AC 8 ms
6,456 KB
testcase_30 AC 12 ms
10,156 KB
testcase_31 AC 3 ms
4,376 KB
testcase_32 AC 2 ms
4,380 KB
testcase_33 AC 7 ms
6,476 KB
testcase_34 AC 7 ms
6,524 KB
testcase_35 AC 12 ms
10,152 KB
testcase_36 AC 5 ms
4,864 KB
testcase_37 AC 13 ms
10,204 KB
testcase_38 AC 13 ms
10,200 KB
testcase_39 AC 12 ms
10,148 KB
testcase_40 AC 1 ms
4,380 KB
testcase_41 AC 1 ms
4,376 KB
testcase_42 AC 13 ms
10,136 KB
testcase_43 AC 12 ms
10,124 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#line 1 "prime_util.test.cpp"
#define PROBLEM "https://yukicoder.me/problems/no/843"
#line 2 "/Users/kodamankod/Desktop/CppProcon/Library/proconlib/utility/int_alias.cpp"
#include <cstdint>
#include <cstddef>

using i32 = std::int32_t;
using u32 = std::uint32_t;
using i64 = std::int64_t;
using u64 = std::uint64_t;
using i128 = __int128_t;
using u128 = __uint128_t;
using isize = std::ptrdiff_t;
using usize = std::size_t;
#line 3 "/Users/kodamankod/Desktop/CppProcon/Library/proconlib/utility/rep.cpp"
#include <algorithm>

class rep {
    struct Iter {
        usize itr;
        constexpr Iter(const usize pos) noexcept: itr(pos) { }
        constexpr void operator ++ () noexcept { ++itr; }
        constexpr bool operator != (const Iter& other) const noexcept { return itr != other.itr; }
        constexpr usize operator * () const noexcept { return itr; }
    };
    const Iter first, last;
public:
    explicit constexpr rep(const usize first, const usize last) noexcept: first(first), last(std::max(first, last)) { }
    constexpr Iter begin() const noexcept { return first; }
    constexpr Iter end() const noexcept { return last; }
};
#line 3 "/Users/kodamankod/Desktop/CppProcon/Library/proconlib/bit/ceil_log2.cpp"

constexpr u64 ceil_log2(const u64 x) {
    u64 e = 0;
    while (((u64) 1 << e) < x) ++e;
    return e;
}
#line 4 "/Users/kodamankod/Desktop/CppProcon/Library/proconlib/utility/auto_realloc.cpp"
#include <utility>
#include <vector>

template <class F>
class AutoRealloc {
    using R = typename decltype(std::declval<F>().operator()(std::declval<usize>()))::value_type;
    F func;
    std::vector<R> data;
public:
    explicit AutoRealloc(F&& func): func(std::forward<F>(func)), data() { }
    explicit AutoRealloc(F&& func, const usize capacity): func(std::forward<F>(func)) { reserve(capacity); }
    void reserve(const usize size) {
        if (data.size() < size) {
            const usize pow2 = ((usize) 1 << ceil_log2(size));
            data = func(pow2);
        }
    }
    R operator [] (const usize i) {
        reserve(i + 1);
        return data[i];
    }
};
#line 7 "/Users/kodamankod/Desktop/CppProcon/Library/proconlib/math/prime_util.cpp"
#include <numeric>
#include <cassert>

struct PrimeUtil {
    static inline auto min_prime = AutoRealloc([](const usize n) {
        std::vector<usize> ret(n);
        std::iota(ret.begin(), ret.end(), (usize) 0);
        std::vector<usize> list;
        for (const usize i: rep(2, n)) {
            if (ret[i] == i) list.push_back(i);
            for (const usize p: list) {
                if (p * i >= n || p > ret[i]) break;
                ret[p * i] = p;
            }
        }
        return ret;
    });
    static bool is_prime(const usize n) {
        if (n <= 1) return false;
        return min_prime[n] == n;
    }
    template <class T>
    static std::vector<std::pair<T, usize>> factorize(T x) {
        assert(x > 0);
        std::vector<std::pair<T, usize>> ret;
        while (x != 1) {
            const usize p = min_prime[x];
            ret.emplace_back((T) p, 0);
            while (min_prime[x] == p) {
                ret.back().second++;
                x /= p;
            }
        }
        return ret;
    }
};
#line 5 "prime_util.test.cpp"
#include <iostream>

int main() {
    u32 N;
    std::cin >> N;
    if (N == 1) {
        std::cout << 0 << '\n';
        return 0;
    }
    u32 ans = 1;
    for (const u32 r: rep(3, N + 1)) {
        if (r * r - 2 > N) break;
        if (PrimeUtil::is_prime(r) && PrimeUtil::is_prime(r * r - 2)) {
            ans += 2;
        }
    }
    std::cout << ans << '\n';
    return 0;
}
0