[SDOI2014]数表

Description
有一张 n×m 的数表，其第 i 行第 j 列（1 <= i <= n, 1 <= j <= m）的数值为
能同时整除 i 和 j 的所有自然数之和。给定 a , 计算数表中不大于 a 的数之和。

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

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

Sample Input
2
4 4 3
10 10 5

Sample Output
20
148

---

首先我们不考虑\(a\)的限制，那么题目要求

\[\sum\limits_{i=1}^n\sum\limits_{j=1}^m\sigma_1(\gcd(i,j))\]

我们直接枚举约数有

\[\sum\limits_{d=1}^n\sum\limits_{i=1}^{\lfloor\frac{n}{d}\rfloor}\sum\limits_{j=1}^{\lfloor\frac{m}{d}\rfloor}\sigma_1(d)[\gcd(i,j)=1]\]

然后我们把\(\sigma_1\)挪到前面，对最后那个式子反演一下有

\[\sum\limits_{d=1}^n\sigma_1(d)\sum\limits_{i=1}^{\lfloor\frac{n}{d}\rfloor}\sum\limits_{j=1}^{\lfloor\frac{m}{d}\rfloor}\sum\limits_{x|i,x|j}\mu(x)\]

把\(x\)挪到前面

\[\sum\limits_{d=1}^n\sigma_1(d)\sum\limits_{x=1}^{\lfloor\frac{n}{d}\rfloor}\mu(x)\lfloor\dfrac{n}{dx}\rfloor\lfloor\dfrac{m}{dx}\rfloor\]

我们令\(T=dx\)，然后更改枚举顺序有

\[\sum\limits_{T=1}^n\lfloor\dfrac{n}{T}\rfloor\lfloor\dfrac{m}{T}\rfloor\sum\limits_{d|T}\sigma_1(d)\mu(\dfrac{T}{d})\]

没有\(a\)的限制这题就做完了……但是现实非常骨感

我们设\(g(T)=\sum\limits_{d|T}\sigma_1(d)\mu(\dfrac{T}{d})\)，可以发现当\(\sigma_1(d)\leqslant a\)时，才会对\(g(T)\)产生贡献

于是我们将询问按\(a\)从小到大排序，枚举询问的时候，\(a\)变大会使得一些\(\sigma_1(d)\)对\(g(T)\)产生贡献，我们就用枚举倍数的方法来找到所有的\(T\)，然后我们需要动态修改\(g(T)\)的值，而且还要支持区间询问，因此我们使用常数较小的树状数组实现

假定所有的\(\sigma_1(d)\)都能产生贡献，枚举所有倍数的复杂度为\(\sum\limits_{i=1}^n\dfrac{n}{i}\thickapprox n\ln n\)，每次更新\(g(T)\)复杂度为\(\log n\)，则修改复杂度为\(O(n\log^2n)\)，每次询问需要数论分块，查询区间和复杂度为\(O(\log n)\)，所以总复杂度为\(O(q\sqrt n\log n+n\log^2n)\)

/*program from Wolfycz*/
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define Fi first
#define Se second
#define MK make_pair
#define inf 0x7f7f7f7f
#define lowbit(x) ((x)&-(x))
using namespace std;
typedef long long ll;
typedef unsigned int ui;
typedef pair<int,int> pii;
typedef unsigned long long ull;
inline char gc(){
    static char buf[1000000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,1000000,stdin),p1==p2)?EOF:*p1++;
}
inline int frd(){
    int x=0,f=1; char ch=gc();
    for (;ch<'0'||ch>'9';ch=gc())   if (ch=='-')    f=-1;
    for (;ch>='0'&&ch<='9';ch=gc()) x=(x<<3)+(x<<1)+ch-'0';
    return x*f;
}
inline int read(){
    int x=0,f=1; char ch=getchar();
    for (;ch<'0'||ch>'9';ch=getchar())  if (ch=='-')    f=-1;
    for (;ch>='0'&&ch<='9';ch=getchar())    x=(x<<3)+(x<<1)+ch-'0';
    return x*f;
}
inline void print(int x){
    if (x<0)    putchar('-'),x=-x;
    if (x>9)    print(x/10);
    putchar(x%10+'0');
}
const int N=1e5;
int mu[N+10],prime[N+10],g[N+10];//$f(d)=\sum\limits_{x|d}x$
bool inprime[N+10];
pii f[N+10];
void prepare(){
    mu[1]=1; int tot=0;
    f[1]=MK(1,1);
    for (int i=2;i<=N;i++){
        if (!inprime[i])    prime[++tot]=i,mu[i]=-1,g[i]=i+1,f[i]=MK(i+1,i);
        for (int j=1;j<=tot&&i*prime[j]<=N;j++){
            inprime[i*prime[j]]=1;
            if (i%prime[j]==0){
                mu[i*prime[j]]=0;
                g[i*prime[j]]=g[i]*prime[j]+1;
                f[i*prime[j]]=MK(f[i].Fi/g[i]*g[i*prime[j]],i*prime[j]);
                break;
            }
            mu[i*prime[j]]=-mu[i];
            f[i*prime[j]]=MK(f[i].Fi*f[prime[j]].Fi,i*prime[j]);
            g[i*prime[j]]=prime[j]+1;
        }
    }
}
int tree[N+10];
void Modify(int x,int v){for (;x<=N;x+=lowbit(x))   tree[x]+=v;}
int Query(int x){
    int res=0;
    for (;x;x-=lowbit(x))   res+=tree[x];
    return res;
}
const int M=2e4;
struct S1{
    int n,m,a,ID;
    void rd(int i){n=read(),m=read(),a=read(),ID=i;}
    bool operator <(const S1 &tis)const{return a<tis.a;}
}A[M+10];
int solve(int n,int m){
    if (n>m)    swap(n,m);
    int res=0;
    for (int i=1,pos;i<=n;i=pos+1){
        pos=min(n/(n/i),m/(m/i));
        res=(res+(Query(pos)-Query(i-1))*(n/i)*(m/i));
    }
    return res;
}
int Ans[M+10];
int main(){
    prepare();
    sort(f+1,f+1+N);
    int Q=read();
    for (int i=1;i<=Q;i++)  A[i].rd(i);
    sort(A+1,A+1+Q);
    for (int i=1,j=1;i<=Q;i++){
        while (f[j].Fi<=A[i].a&&j<=N){
            for (int k=f[j].Se;k<=N;k+=f[j].Se)
                Modify(k,f[j].Fi*mu[k/f[j].Se]);
            j++;
        }
        Ans[A[i].ID]=solve(A[i].n,A[i].m);
    }
    for (int i=1;i<=Q;i++)  printf("%d\n",Ans[i]&(~(1<<31)));
    return 0;
}

转载于:https://www.cnblogs.com/Wolfycz/p/10275666.html