[bzoj3529][SDOI2014]数表

3529: [Sdoi2014]数表

Time Limit: 10 Sec Memory Limit: 512 MB
Submit: 991 Solved: 522
[Submit][Status][Discuss]
Description

有一张N×m的数表，其第i行第j列（1 < =i < =礼，1 < =j < =m）的数值为

能同时整除i和j的所有自然数之和。给定a，计算数表中不大于a的数之和。

Input

输入包含多组数据。
输入的第一行一个整数Q表示测试点内的数据组数，接下来Q行，每行三个整数n，m，a(|a| < =10^9)描述一组数据。

Output

对每组数据，输出一行一个整数，表示答案模2^31的值。

Sample Input

2

4 4 3

10 10 5

Sample Output

20

148

HINT

1 < =N．m < =10^5  ， 1 < =Q < =2×10^4

先不考虑a的限制，那么式子就可以先写成这样

$$\sum_{i=1}^{n}\sum_{j=1}^{m}\sum_{d|i,d|j}d$$
$$设f(x)表示x的约数和$$
$$\sum_{i=1}^{n}\sum_{j=1}^{m}\sum_{d=1}^{min(n,m)}f(d)[gcd(i,j)==d]$$
$$\sum_{d=1}^{min(n,m)}f(d)\sum_{i=1}^{\lfloor\frac{n}{d}\rfloor}\sum_{j=1}^{\lfloor\frac{m}{d}\rfloor}[gcd(i,j)==1]$$
$$\sum_{d=1}^{min(n,m)}f(d)\sum_{i=1}^{\lfloor\frac{n}{d}\rfloor}\sum_{j=1}^{\lfloor\frac{m}{d}\rfloor}\sum_{p|i,p|j}\mu(p)$$
$$\sum_{d=1}^{min(n,m)}f(d)\sum_{p=1}^{min(\lfloor\frac{n}{d}\rfloor,\lfloor\frac{m}{d}\rfloor)}\lfloor\frac{\lfloor\frac{n}{d}\rfloor}{p}\rfloor\lfloor\frac{\lfloor\frac{m}{d}\rfloor}{p}\rfloor\mu(p)$$
$$\sum_{d=1}^{min(n,m)}f(d)\sum_{p=1}^{min(\lfloor\frac{n}{d}\rfloor,\lfloor\frac{m}{d}\rfloor)}\lfloor\frac{n}{dp}\rfloor\lfloor\frac{m}{dp}\rfloor\mu(p)$$
$$设t=dp$$
$$\sum_{t=1}^{min(n,m)}\lfloor\frac{n}{t}\rfloor\lfloor\frac{m}{t}\rfloor\sum_{d|t}f(d)\mu(\frac{t}{d})$$
$f()$是一个极性函数，这样后面的 $f()\mu()$预处理前缀和，前面的分块，复杂度就是 $O(\sqrt{n})$的。
现在看如何加上a这个限制。
我们可以离线，讲询问按照a从小到大排序，然后再将f按照从小到大排序。这样询问的时候，就可以动态的去维护一个树状数组。
对于每一个询问，将 $f(d)$小于等于当前询问的a的每一个 $f()\mu()$插入到树状数组的对应位置上就好了。
复杂度是 $O(\sqrt{n}logn)$

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define D 2147483647
const int N=100000;
bool flag[N+10];
struct F{int d,No;}f[N+10];
struct S{int n,m,a,No,ans;}q[N+10];
int T,prime[N+10],u[N+10],tr[N+10],g[N+10],sum[N+10];
inline bool cmp_1(S x,S y){return x.a<y.a;}
inline bool cmp_2(S x,S y){return x.No<y.No;}
inline bool cmp_3(F x,F y){return x.d<y.d;}
inline int in(){
    int x=0;char ch=getchar();
    while(ch<'0'||ch>'9') ch=getchar();
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x;
}
inline int quickpow(int x,int y){
    int ans=1;
    while(y){
        if(y&1) ans=ans*x;
        y>>=1; x=x*x;
    }
    return ans;
}
inline void prepare(){
    int i,j;
    f[1].d=f[1].No=u[1]=1;
    for(i=2;i<=N;++i){
        f[i].No=i;
        if(!flag[i]){
            prime[++prime[0]]=i;
            u[i]=-1;
            f[i].d=i+1;
            g[i]=1;
            sum[i]=1+i;
        }
        for(j=1;j<=prime[0]&&i*prime[j]<=N;++j){
            flag[i*prime[j]]=true;
            if(i%prime[j]==0){
                u[i*prime[j]]=0;
                g[i*prime[j]]=g[i]+1;
                sum[i*prime[j]]=sum[i]+quickpow(prime[j],g[i]+1);
                f[i*prime[j]].d=f[i].d/sum[i]*sum[i*prime[j]];
                break;
            }
            u[i*prime[j]]=-u[i];
            f[i*prime[j]].d=f[i].d*f[prime[j]].d;
            g[i*prime[j]]=1;
            sum[i*prime[j]]=prime[j]+1;
        }
    }
}
inline void insert(int x,int y){
    for(;x<=N;x+=x&(-x)) tr[x]+=y;
}
inline int query(int x){
    int sum=0;
    for(;x;x-=x&(-x)) sum+=tr[x];
    return sum;
}
int main(){
    prepare();
    int i,j,now=1,pos;
    T=in();
    for(i=1;i<=T;++i){
        q[i].n=in();q[i].m=in();q[i].a=in();
        q[i].No=i;
    }
    sort(q+1,q+T+1,cmp_1);
    sort(f+1,f+N+1,cmp_3);
    for(i=1;i<=T;++i){
        while(f[now].d<=q[i].a&&now<=N){
            for(j=1;j*f[now].No<=N;++j) 
              insert(j*f[now].No,f[now].d*u[j]);
            ++now;
        }
        if(q[i].n>q[i].m) swap(q[i].n,q[i].m);
        for(j=1;j<=q[i].n;j=pos+1){
            pos=min(q[i].n/(q[i].n/j),q[i].m/(q[i].m/j));
            q[i].ans+=(q[i].n/j)*(q[i].m/j)*(query(pos)-query(j-1));
            q[i].ans&=D;
        }
    }
    sort(q+1,q+T+1,cmp_2);
    for(i=1;i<=T;++i) printf("%d\n",q[i].ans);
}