2019CCPC网络赛 HD6707——杜教筛

题意

求 $f(n,a,b)=\sum_{i=1}^n \sum_{j=1}^i gcd(i^a-j^a,i^b-j^b)[gcd(i,j)=1]\%(10^9+7)$,$1 \le n,a,b \le 10^9$,共有 $T$ 组测试,其中只有10组的 $n$ 大于 $10^6$.

分析

首先,当 $i, j$互质,$a, b$互质时,有 $gcd(i^a-j^a,i^b-j^b)=i-j$(证明见 链接),也可以打表猜一猜嘛。

可以推出:$$\sum_{d=1}^{N}\mu(d)\cdot d\sum_{i=1}^{\lfloor\frac{N}{d}\rfloor}\sum_{j=1}^{i}(i-j)$$

单独考虑后半部分,$\sum_{i=1}^{k}\sum_{j=1}^{i}(i-j)=\frac{k^3-k}{6}$.

然后,只剩下左边的 $\mu(d)\cdot d$,

将其与恒等函数 $Id(n) = n$ 狄利克雷卷积后得

$$\begin{align*}
(\mu(d)\cdot d)*Id(d)
& =  \sum_{d|n}(\mu(d)\cdot d)\cdot Id(\frac{n}{d})\\
& = \sum_{d|n}\mu(d) =  [n=1]
\end{align*}$$

接下来套杜教筛得公式

$$\begin{align*}
S(n)
& = \sum\limits_{i=1}^n [i=1]-\sum\limits_{i=2}^ni\cdot S(\lfloor\dfrac{n}{i}\rfloor)\\
& = 1-\sum\limits_{i=2}^ni\cdot S(\lfloor\dfrac{n}{i}\rfloor)
\end{align*}$$

#include
using namespace std;

typedef long long ll;
const int maxn = 6e6 + 10;
const ll mod = 1e9+7;
const ll inv6 = 166666668;
int sum[maxn], mu[maxn], pri[maxn], pn;
bool vis[maxn];
map<int, int>mp_sum;
int n, a, b;

ll s2(ll i, ll j)
{
    return (i+j) * (j-i+1) / 2 % mod;
}

ll s3(ll k)
{
    return (k*k%mod - 1) * k % mod * inv6 % mod;
}

ll S(ll x)
{
    if(x < maxn)  return sum[x];
    if(mp_sum[x])  return mp_sum[x];
    ll ret = 1LL;
    for(int i = 2, j;i <= x;i = j+1)
    {
        j = x / (x / i);
        ret = (ret - s2(i, j) * (S(x/i))%mod) % mod;
    }
    return  mp_sum[x] = (ret + mod) % mod;
}

void pre()
{
    mu[1] = 1;
    for(int i = 2;i < maxn;i++)
    {
        if(!vis[i])
        {
            pri[++pn] = i;
            mu[i] = -1;
        }
        for(int j = 1;j <= pn && i * pri[j] < maxn; j++)
        {
            vis[i * pri[j]] = true;
            if(i % pri[j])  mu[i * pri[j]] = -mu[i];
            else
            {
                mu[i * pri[j]] = 0;
                break;
            }
        }
    }
    for(int i = 1;i < maxn;i++)  sum[i] = (sum[i-1] + i * mu[i]) % mod;
}

int main()
{
    pre();

    int T;
    scanf("%d", &T);
    while(T--)
    {
        scanf("%d%d%d", &n, &a, &b);
        ll ans  = 0;
        for(ll l = 1,r; l <= n;l = r+1)
        {
            r = n / (n / l);
            ans = (ans + (S(r) - S(l-1)) * s3(n/l)) % mod;
        }
        printf("%lld\n", (ans+mod)%mod);
    }
    return 0;
}

最开始开的 MAXN=2e6,会TLE;原博客开的6e6,又MLE,将long long 数组改成 int 才行。

其实标答是推成 $\displaystyle ans = \frac{\sum _{i=1}^n i\varphi (i) - 1}{2}$,少了一次整除分块。

但是,通过这种解法,让我深刻认识了杜教筛的时空矛盾该怎么平衡。

 

参考链接:https://segmentfault.com/a/119000002017183

转载于:https://www.cnblogs.com/lfri/p/11410031.html

你可能感兴趣的:(2019CCPC网络赛 HD6707——杜教筛)