bzoj4173 数学

Description

---

N,M<=10^15

Solution

---

首先需要知道一个结论，若$n \% k+m \% k\geq k$ ，则$\lfloor\frac{n+m}{k}\rfloor-\lfloor\frac{n}{k}\rfloor-\lfloor\frac{m}{k}\rfloor=1$
证明如下：
显然可以设$n=a_1k+r_1$，$m=a_2k+r_2$，那么$\lfloor\frac{n+m}{k}\rfloor=a_1+a_2+1$，$\lfloor\frac{n}{k}\rfloor=a_1$，$\lfloor\frac{m}{k}\rfloor=a_2$
因为$r_1+r_2\geq k$因此$\lfloor\frac{r_1+r_2}{k}\rfloor=1$

现在令$t=\sum_{n\%k+m\%k\geq k}\varphi(k)$，显然答案就是$\varphi(n)*\varphi(m)*t$
那么有$t=\sum_{n\%k+m\%k\geq k}\varphi(k)=\sum_{k=1}^{n+m}\varphi(k)[\lfloor\frac{n+m}{k}\rfloor-\lfloor\frac{n}{k}\rfloor-\lfloor\frac{m}{k}\rfloor=1]$
我们发现后面那一坨布尔表达式不为0就为1，那么有$t=\sum_{k=1}^{n+m}\varphi(k)(\lfloor\frac{n+m}{k}\rfloor-\lfloor\frac{n}{k}\rfloor-\lfloor\frac{m}{k}\rfloor)$，可以感受一下区别
那么拆一下就成了$t=\sum_{k=1}^{n+m}\varphi(k)\lfloor\frac{n+m}{k}\rfloor-\sum_{k=1}^{n}\varphi(k)\lfloor\frac{n}{k}\rfloor-\sum_{k=1}^{m}\varphi(k)\lfloor\frac{m}{k}\rfloor$

考虑一下$\sum_{k=1}^{n}\varphi(k)\lfloor\frac{n}{k}\rfloor$的意义，有$\sum_{k=1}^{n}\varphi(k)\lfloor\frac{n}{k}\rfloor=\sum_{k=1}^{n}\sum_{i=1}^{\lfloor\frac{n}{k}\rfloor}\varphi(k)=\sum_{i=1}^{n}\sum_{k|i}\varphi(k)=\sum_{i=1}^{n}i$
辣莫$t=\frac{1}{2}[(n+m)(n+m+1)-n(n+1)-m(m+1)]=nm$
即$ans=\varphi(n)*\varphi(m)*nm$
惊喜不惊喜，意外不意外

Code

---

#include <stdio.h>
#include <string.h>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)

typedef long long LL;
const int MOD=998244353;

void cal(LL &ret,LL x) {
    ret=ret/x;
    ret=ret*(x-1);
}

LL get_phi(LL n) {
    LL ret=n;
    for (LL i=2;i*i<=n;i++) {
        if (n%i) continue;
        cal(ret,i);
        while (n%i==0) n/=i;
    }
    if (n!=1) ret=(ret/n)*(n-1);
    return ret;
}

int main(void) {
    LL n,m; scanf("%lld%lld",&n,&m);
    printf("%lld\n",n%MOD*m%MOD*(get_phi(m)%MOD)%MOD*(get_phi(n)%MOD)%MOD);
    return 0;
}