結果

問題 No.1653 Squarefree
ユーザー 👑 NachiaNachia
提出日時 2024-01-28 17:54:14
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,849 ms / 2,000 ms
コード長 3,252 bytes
コンパイル時間 679 ms
コンパイル使用メモリ 69,248 KB
実行使用メモリ 82,152 KB
最終ジャッジ日時 2024-09-28 09:55:30
合計ジャッジ時間 55,574 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1,787 ms
81,380 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 1 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 1 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 2 ms
5,376 KB
testcase_13 AC 1,840 ms
82,148 KB
testcase_14 AC 1,849 ms
81,892 KB
testcase_15 AC 1,810 ms
82,020 KB
testcase_16 AC 1,823 ms
81,760 KB
testcase_17 AC 1,807 ms
81,892 KB
testcase_18 AC 1,794 ms
81,888 KB
testcase_19 AC 1,830 ms
81,892 KB
testcase_20 AC 1,790 ms
82,152 KB
testcase_21 AC 1,805 ms
82,016 KB
testcase_22 AC 1,832 ms
82,020 KB
testcase_23 AC 1,804 ms
82,020 KB
testcase_24 AC 1,799 ms
81,256 KB
testcase_25 AC 1,833 ms
81,764 KB
testcase_26 AC 1,793 ms
82,148 KB
testcase_27 AC 1,829 ms
81,764 KB
testcase_28 AC 1,652 ms
76,644 KB
testcase_29 AC 1,775 ms
82,020 KB
testcase_30 AC 1,831 ms
82,020 KB
testcase_31 AC 1,767 ms
82,148 KB
testcase_32 AC 1,757 ms
82,020 KB
testcase_33 AC 1,786 ms
82,020 KB
testcase_34 AC 1,775 ms
82,012 KB
testcase_35 AC 1,790 ms
82,024 KB
testcase_36 AC 1,788 ms
81,892 KB
testcase_37 AC 1,732 ms
81,768 KB
testcase_38 AC 1,774 ms
81,892 KB
testcase_39 AC 1,776 ms
81,676 KB
testcase_40 AC 1,772 ms
81,764 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <vector>
#include <cmath>

#include <cassert>

namespace nachia{

namespace internal{

// mod 2^64
constexpr unsigned long long PowerOfULongLong(unsigned long long a, unsigned long long i){
    unsigned long long res = 1;
    while(i){ if(i&1){ res *= a; } i /= 2; a *= a; }
    return res;
}

}

unsigned long long FloorOfKthRoot(unsigned long long real, unsigned long long k){
    using u64 = unsigned long long;
    assert(k != 0);
    if(real <= 1) return real;
    if(k >= 64) return 1;
    if(k == 1) return real;
    struct Precalc{
        // a^i <= x
        static constexpr bool lesseq(u64 a, int i, u64 x) {
            if (a == 0) return true;
            for(int j=0; j<i; j++) x /= a;
            return x >= 1;
        }
        unsigned long long BORDER[64];
        constexpr Precalc() : BORDER() {
            for (int idx = 2; idx <= 63; idx++) {
                u64 l = 0, r = 1ull << 33;
                while (l + 1 < r) {
                    u64 m = (l + r) / 2;
                    if (lesseq(m, idx, ~0ull)) l = m;
                    else r = m;
                }
                BORDER[idx] = r;
            }
        };
    };
    constexpr Precalc precalc;
    u64 l = 0, r = precalc.BORDER[k];
    if(real < r) r = real;
    while (l + 1 < r) {
        u64 m = (l + r) / 2;
        if(internal::PowerOfULongLong(m, k) <= real) l = m;
        else r = m;
    }
    return l;
}

unsigned long long CeilOfKthRoot(unsigned long long real, unsigned long long k){
    if(real <= 1) return real;
    if(k >= 64) return 2;
    if(k == 1) return real;
    unsigned long long x = FloorOfKthRoot(real, k);
    if(internal::PowerOfULongLong(x, k) != real) x++;
    return x;
}

} // namespace nachia

namespace nachia{

long long CountSquarefree(long long n){
    using i64 = long long;
    i64 s = 0;
    auto getMobius = [&](i64 n){
        std::vector<bool> sieve(n+1, true);
        std::vector<signed char> mu(n+1, 1);
        for(i64 i=2; i<=n; i++) if(sieve[i]){
            mu[i] = -1;
            for(i64 j=i*i; j<=n; j+=i) sieve[j] = false;
            for(i64 j=i*2; j<=n; j+=i) mu[j] = -mu[j];
            for(i64 j=i*i; j<=n; j+=i*i) mu[j] = 0;
        }
        return mu;
    };
    if(n <= 4000){
        auto mu = getMobius(n+1);
        for(i64 i=1; i*i<=n; i++) s += n/(i*i) * mu[i];
        return s;
    }
    i64 I = nachia::FloorOfKthRoot(n, 5);
    i64 D = nachia::FloorOfKthRoot(n/(I+1), 2);
    auto mu = getMobius(D+1);
    std::vector<int> Mf(D+1);
    for(i64 i=1; i<=D; i++) Mf[i] = Mf[i-1] + mu[i];
    std::vector<i64> Md(I+1);
    for(i64 i=I; i>=1; i--){
        i64 m = 1;
        i64 x = nachia::FloorOfKthRoot(n/i, 2);
        i64 Dx = nachia::FloorOfKthRoot(x, 2);
        i64 Rx = x / (Dx+1);
        i64 r = 2;
        for( ; i*r*r<=I; r++) m -= Md[i*r*r];
        for( ; r<=Rx; r++) m -= Mf[x/r];
        for(i64 d=1; d<=Dx; d++) m -= mu[d] * (x/d - Rx);
        Md[i] = m; s += m;
    }
    for(i64 i=1; i<=D; i++) s += mu[i] * (n/(i*i) - I);
    return s;
}

} // namespace nachia
#include <cstdio>

int main(){
    long long L, R; scanf("%lld%lld", &L, &R);
    long long ans = nachia::CountSquarefree(R) - nachia::CountSquarefree(L-1);
    printf("%lld\n", ans);
    return 0;
}
0