bzoj3529 [Sdoi2014]数表（反演）

题目链接

分析：
题目所求：

$$\sum_{i=1}^{n}\sum_{j=1}^{m}h(i,j)[h(i,j)<=a],h(i,j)=能同时整除i和j的所有自然数之和$$

实际上，$h$的意义：$gcd(i,j)$的约数和
设$f(n)$是$n$的约数和

随意查了一下，约数和定理：
$n=p_1^{a_1}*p_2^{a_2}*p_3^{a_3}*…*p_k^{a_k}$
$f(n)=(p_0^0+p_1^1+…p_1^{a_1})(p_2^0+p_2^1+…p_2^{a_2})…(p_k^0+p_k^1+…p_k^{a_k})$

我们发现当$p,q$互质时，$f(pq)=f(p)f(q)$，$f$是个积性函数
也就是说$f$可以线性筛求解（当然直接暴力求也没什么影响）

$a$的限制比较复杂，简单的，我们先来看看如果没有$a$的限制，是个什么情况

$$\sum_{k=1}^{min(n,m)}\sum_{i=1}^{n}\sum_{j=1}^{m}f(k)[gcd(i,j)=k]$$

设$g(i)$表示$\sum_{x=1}^{n}\sum_{y=1}^{m}[gcd(x,y)=i]$

$$g(i)=\sum_{d=1}^{min(n/i,m/i)}{n \over di}{m \over di}μ(d)$$

则原式：

$$\sum_{k=1}^{min(n,m)}f(k)\sum_{d=1}^{min(n/k,m/k)}{n \over dk}{m \over dk}μ(d)$$

设$Q=dk$

$$\sum_{k=1}^{min(n,m)}f(k)\sum_{Q=1}^{min(n,m)}{n \over Q}{m \over Q}μ({Q \over k})=\sum_{Q=1}^{min(n,m)}{n \over Q}{m \over Q}\sum_{k|Q}f(k)μ({Q \over k})$$

显然$\sum_{k|Q}f(k)μ({Q \over k})$是$f$函数与$μ$函数的狄利克雷卷积
预处理前缀和，前面的${n \over Q}{m \over Q}$分块，那么这个式子的计算就达到了$\sqrt n$

那么我们再加上$a$的限制
离线，将询问按照$a$从小到大排序，$f$函数按照从小到大排序
询问的时候，我们就动态维护一个树状数组$c$
每次处理询问之前，把数值小于当前a的$f$值加入树状数组
树状数组的每一位$c[x]$存的$\sum_{d|x}f(d)μ({x \over d})$
数据组数为$T$，询问复杂度$O(T\sqrt{n}logn)$

注意：

这题复杂度有点卡，对于取模运算我们可以对$int$自然溢出，最后再$\&2147483647$就可以了

tip

PoPoQQQ大大的题解

第一次写完，发现小数据OK，大数据WA了（比标准答案大一点）、
竟然是线性筛mu写错了。。。orz
注意最后的取模，一不小心就会炸掉变成负数

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>

using namespace std;

const int N=100005;
int n,m,Q,sshu[N],tot=0,mu[N],ans[N],mx,c[N];
bool no[N];
struct node{
    int n,m,A,id;
    bool operator <(const node &b) const {
        return A<b.A;
    }
};
node a[N];
struct node2{
    int x,id;
    bool operator <(const node2 &b) const {
        return x<b.x||(x==b.x&&id<b.id);
    }
};
node2 f[N];

void prepare(int n) {
    memset(no,0,sizeof(no));
    mu[1]=1;
    for (int i=2;i<=n;i++) {
        if (!no[i]) {
            sshu[++tot]=i;
            mu[i]=-1; 
        }
        for (int j=1;j<=tot&&sshu[j]*i<=n;j++) {
            no[i*sshu[j]]=1;
            if (i%sshu[j]==0) {
                mu[i*sshu[j]]=0;
                break;
            }
            mu[i*sshu[j]]=-mu[i];
        }
    }
    for (int i=1;i<=n;i++)        //暴力求f 
        for (int j=i;j<=n;j+=i)
            f[j].x+=i;
    for (int i=1;i<=n;i++) f[i].id=i;
}

void add(int x,int z) {for (int i=x;i<=mx;i+=(i&(-i))) c[i]+=z;}
int ask(int x) {int ans=0; for (int i=x;i>0;i-=(i&(-i))) ans+=c[i]; return ans;} 

void solve(int _) {
    int n=a[_].n,m=a[_].m;
    int last=0;
    for (int i=1;i<=min(n,m);i=last+1) {
        last=min(n/(n/i),m/(m/i));
        ans[a[_].id]+=(n/i)*(m/i)*(ask(last)-ask(i-1));
    }
}

int main()
{
    mx=0;
    scanf("%d",&Q);
    for (int i=1;i<=Q;i++) {
        scanf("%d%d%d",&a[i].n,&a[i].m,&a[i].A);
        a[i].id=i; mx=max(mx,min(a[i].n,a[i].m));
    }
    prepare(mx);
    sort(a+1,a+1+Q);
    sort(f+1,f+1+mx);
    int now=1;
    for (int i=1;i<=Q;i++) {
        while (now<=mx&&f[now].x<=a[i].A) {
            for (int j=f[now].id;j<=mx;j+=f[now].id)   //sum(k|Q) f(k)mu(Q/k) 所有k的倍数Q都需要add 
                add(j,f[now].x*mu[j/f[now].id]);       //卷积 
            now++;
        }
        solve(i);
    }
    for (int i=1;i<=Q;i++) printf("%d\n",ans[i]&((1<<31)-1));
    return 0;
}