首页 > 其他 > 详细

【数学】质数的k次方和

时间:2020-09-21 08:28:01      阅读:58      评论:0      收藏:0      [点我收藏+]

质数的1次方和,也就是质数求和。

int mod;

inline ll add_mod(ll x, ll y) {
    return (x + y >= mod) ? (x + y - mod) : (x + y);
}

inline ll sub_mod(ll x, ll y) {
    return (x < y) ? (x - y + mod) : (x - y);
}

inline ll mul_mod(ll x, ll y) {
    return x * y % mod;
}

inline ll pow_mod(ll x, ll y) {
    ll t = 1;
    while(y) {
        if(y & 1)
            t = mul_mod(t, x);
        x = mul_mod(x, x);
        y >>= 1;
    }
    return t;
}

inline ll sum(ll n) {
    n %= mod;
    return (n * (n + 1)) / 2 % mod;
}

const int MAXN = 1e6 + 5;

ll ssum[MAXN];
ll lsum[MAXN];
bool mark[MAXN];

ll prime_sum(ll n) {
    const int v = sqrt(n);
    ssum[0] = 0;
    lsum[0] = 0;
    memset(mark, 0, sizeof(mark[0]) * (v + 1));
    for(ll i = 1; i <= v; ++i) {
        ssum[i] = sum(i) - 1;
        lsum[i] = sum(n / i) - 1;
    }
    for(ll p = 2; p <= v; ++p) {
        if(ssum[p] == ssum[p - 1])
            continue;
        ll psum = ssum[p - 1];
        ll q = p * p;
        ll ed = min((ll)v, n / q);
        ll delta1 = (p & 1) + 1;
        for(ll i = 1; i <= ed; i += delta1) {
            if(!mark[i]) {
                ll d = i * p;
                ll tmp = (d <= v) ? lsum[d] : ssum[n / d];
                tmp = sub_mod(tmp, psum);
                tmp = mul_mod(tmp, p);
                lsum[i] = sub_mod(lsum[i], tmp);
            }
        }
        ll delta2 = p * delta1;
        for(ll i = q; i <= ed; i += delta2)
            mark[i] = 1;
        for(ll i = v; i >= q; --i) {
            ll tmp = ssum[i / p];
            tmp = sub_mod(tmp, psum);
            tmp = mul_mod(tmp, p);
            ssum[i] = sub_mod(ssum[i], tmp);
        }
    }
    return lsum[1];
}

【数学】质数的k次方和

原文:https://www.cnblogs.com/purinliang/p/13703156.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!