【hdu 6889】Graph Theory Class (min_25 筛)

abcdhhhh_

已于 2022-08-11 10:32:37 修改

阅读量113

点赞数

分类专栏： ACM

于 2021-09-01 17:19:34 首次发布

本文链接：https://blog.csdn.net/qq_47903865/article/details/120041805

版权

最小生成树素数完全图筛法算法实现

关键词由CSDN通过智能技术生成

ACM 专栏收录该内容

45 篇文章 1 订阅

订阅专栏

题目链接

题目大意

$n$ 个点的完全图，点的编号为 $1\sim n$ ，点 $i$ 和 $j$ 之间的边权为 $\text{lcm}(i+1, j+1)$ ，求MST的大小。（ $n\le 10^{10}$ ）

思路

将点的编号+1。
点2无贡献，其他点编号为素数的贡献为2，其他点贡献为1。
因此关键要求出 $n$ 以内的素数和。
需要用到 min_25 筛。

min_25筛通常用来求积性函数 $f (n)$ 的前缀和，要求是 $f (p)$ 为关于 $p$ 的多项式，且 $f(p^c)$ 可以快速求。

具体做法是先把所有点的取值均当作 $f (p)$ 的多项式，做一个幂和（参考），然后利用 $\sqrt{n}$ 以内的质数，对合数上的值进行修正。可以记 $f_i(n)$ 为筛掉前 $i$ 个质数时的前缀和（可能要搞记忆化）。

回到本题，可以仿照上述过程实现筛法。

代码

#include <bits/stdc++.h>
#define rep(i, l, r) for (int i = l; i <= r; ++i)
typedef long long ll;
const int N = 20000007;
using namespace std;
ll mod;
int t;
int prime[N], tot = 0;
ll sp[N], dp[N];
bool iscmp[N];
void init() {  // get primes
    dp[0] = dp[1] = 0;
    rep(i, 2, N - 1) {
        dp[i] = dp[i - 1];
        if (!iscmp[i]) {
            prime[++tot] = i;
            dp[i] += i;
        }
        for (int j = 1; j <= tot && i * prime[j] < N; ++j) {
            iscmp[i * prime[j]] = true;
            if (i % prime[j] == 0) break;
        }
    }
    rep(i, 1, tot) sp[i] = sp[i - 1] + prime[i];
}
ll n;
ll f(ll u, ll j) {  // g(n/u, j)
    // printf("%d %d\n", u, j);
    ll now = n / u;
    if (now < N && now < 1ll * prime[j] * prime[j]) return dp[now];
    while (j && now < 1ll * prime[j] * prime[j]) --j;  // useless
    if (!j) return (now * (now + 1) / 2 - 1) % mod;    // not removed
    return ((f(u, j - 1) - (f(u * prime[j], j - 1) - sp[j - 1]) * prime[j]) %
                mod +
            mod) %
           mod;
}

int main() {
    init();
    // printf("%d\n", sp[26]+5151-5);
    scanf("%d", &t);
    while (t--) {
        scanf("%lld%lld", &n, &mod);
        ++n;
        ll inv2 = (mod + 1) / 2;
        ll ans = n % mod * ((n + 1) % mod) % mod * inv2 % mod;
        ans = (ans + f(1, tot)) % mod;
        ans = (ans + mod - 5) % mod;
        printf("%lld\n", ans);
    }
    return 0;
}