题意:求
$$\sum_{i=1}^{n} \sum_{j=1}^{m} \sum_{d|(i, j)} d 且 (\sum_{d|(i, j)} d)<=a$$
n, m<=1e5,q次询问,q<=2*1e4
#include <bits/stdc++.h>
using namespace std;
const int N=1e5+10, MN=1e5, YU=(1u<<31)-1;
int c[N], mx;
void upd(int x, int s) { for(; x<=mx; x+=x&-x) c[x]+=s; }
int sum(int x) { int r=0; for(; x; x-=x&-x) r+=c[x]; return r; } bool np[N];
int p[N], pcnt, mu[N], f[N], last[N];
void init() {
mu[1]=f[1]=1;
int i, j;
for(i=2; i<=MN; ++i) {
if(!np[i]) p[++pcnt]=i, mu[i]=-1, f[i]=1+i, last[i]=1;
for(j=1; j<=pcnt; ++j) {
int t=p[j]*i; if(t>MN) break;
np[t]=1;
if(i%p[j]==0) { mu[t]=0; f[t]=last[i]+p[j]*f[i]; last[t]=last[i]; break; }
mu[t]=-mu[i]; f[t]=f[i]*(1+p[j]);
last[t]=f[i];
}
}
}
void update(int d, int s) {
for(int i=d, j=1; i<=mx; i+=d, ++j) upd(i, s*mu[j]);
} struct A { int n, m, a, id; }q[N];
struct B { int f, id; }F[N];
inline bool cmp1(const A &a, const A &b) { return a.a<b.a; }
inline bool cmp2(const B &a, const B &b) { return a.f<b.f; } int Ans[N];
int main() {
init();
int Q; scanf("%d", &Q);
for(int i=1; i<=Q; ++i) scanf("%d %d %d", &q[i].n, &q[i].m, &q[i].a), q[i].id=i, mx=max(mx, max(q[i].n, q[i].m));
sort(q+1, q+1+Q, cmp1);
for(int i=1; i<=mx; ++i) F[i].id=i, F[i].f=f[i];
sort(F+1, F+1+mx, cmp2); int n, m, a, now=1, pos, ans;
for(int k=1; k<=Q; ++k) {
n=q[k].n, m=q[k].m, a=q[k].a; if(n>m) swap(n, m);
while(now<=mx && F[now].f<=a) update(F[now].id, F[now].f), ++now;
pos=ans=0;
for(int i=1; i<=n; i=pos+1) {
pos=min(n/(n/i), m/(m/i));
ans+=(sum(pos)-sum(i-1))*(n/i)*(m/i);
}
Ans[q[k].id]=ans&YU;
}
for(int i=1; i<=Q; ++i) printf("%d\n", Ans[i]);
return 0;
}
题解:
终于会了反演= = 其实不就是一个公式吗= =....我们来膜拜PoPoQQQ (http://wenku.baidu.com/view/fbec9c63ba1aa8114431d9ac.html)
手推了半小时啊= =在推$f(i)$的时候竟然还不知道有约数和的公式这个玩意QAQ,感谢PoPoQQQ对我的教导,然后我根据这个玩意直接线性筛和mu一起预处理出来了...
首先我们变化一下公式,且先不考虑$<=a$这个条件,且下边均默认$n<=m$:
设
$$g(d) = \sum_{i=1}^{n} \sum_{j=1}^{m} [(i, j)=d]$$
$$f(d) = \sum_{i|d} i$$
那么本题可以变为枚举$gcd$,求
$$\sum_{i=1}^{n} g(i)f(i)$$
我们先来求$g(d)$(原来的博文是直接展开求的,现在我们用莫比乌斯反演来做...)
设
$$F(d)表示d|(i, j)的对数,而g(d)是表示(i, j)=d的对数$$
易得
$$ F(d)=
\begin{cases}
\lfloor \frac{n}{d} \rfloor \lfloor \frac{m}{d} \rfloor \\
\sum_{d|i} g(i) \\
\end{cases}
$$
根据莫比乌斯反演(两种形式,约数和 与 倍数和,这里是倍数和的形式,证明略),易得
$$ g(d) = \sum_{d|i} \mu (\frac{i}{d}) F(i) $$
$f(i)$的求法待会再说。那么现在我们展开之前得到的式子
$$\sum_{i=1}^{n} g(i)f(i) = \sum_{i=1}^{n} f(i) \sum_{i|T} \mu (\frac{T}{i}) F(T) = \sum_{i=1}^{n} f(i) \sum_{i|T} \mu (\frac{T}{i}) \lfloor \frac{n}{T} \rfloor \lfloor \frac{m}{T} \rfloor $$
继续换指标,我们来枚举$T$,最终可得
$$\sum_{T=1}^{n} \lfloor \frac{n}{T} \rfloor \lfloor \frac{m}{T} \rfloor \sum_{i|T} f(i) \mu (\frac{T}{i})$$
完美的公式= =
那么我们只需要求出$f(i)$然后按照倍数暴力更新即可,$O(nlnn)$(其实如果本题没有$<=a$这个条件,可以直接线性筛筛出来而不需要暴力枚举,详细请见我之前的博文http://www.cnblogs.com/iwtwiioi/p/4132095.html
那么考虑当有$<=a$这个条件了,发现其实就是$f(i)<=a$,那么发现其实只需要排序一下,然后用个bit动态维护前缀和即可,最终查询用分块。
复杂度$O(nlogn+qn^{0.5}log n)$
最后来说$f(i)$的线性求法...
由于
$$f(i) = \prod_{i} \sum_{j=0}^{a[i]} p_{i}^{j},a[i]表示指数数目$$
我们考虑$f(kp_y)$,即线性筛里边的外层循环$k$和内层循环$p_y$
当$p_y | k$时
$$f(kp_y) = \prod_{i} \sum_{j=0}^{a[i]} p_{i}^{j}$$
我们在$p_y$那个指数提取一个出来,则那一部分的和变成$p_y \sum_{j=-1}^{a[y]-1} p_{y}^{j}$,将指数为$-1$的一起提出来,最终整理得
$$f(kp_y) = \prod_{i \neq y} \sum_{j=0}^{a[i]} p_{i}^{j} + p_y \prod_{i} \sum_{j=0}^{a[i]或当i=y时, a[y]-1} p_{i}^{j} = f(k去除所有p_y因子) + p_y f(k)$$
当$p_y \nmid k$时,好算多了...直接可推得
$$f(kp_y) = (1+p_y)f(k)$$
那么就大胆放到线性筛里去求吧!