【刷题】BZOJ 3529 [Sdoi2014]数表

时间:2021-11-21 15:44:20

Description

有一张n×m的数表,其第i行第j列(1<=i<=n,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

Solution

莫比乌斯反演,膜拜PoPoQQQ

先不管a的限制,我们直接求数表中所有数之和\(ans'\)

设\(F(i)\)为i的约数和,\(g(i)\)为在限制n和m范围内gcd为i的对数的个数,那么

\[ans'=\sum_{i=1}^{min(n,m)}F(i)g(i)
\]

\(g(i)\)的反演已经是老套路了,直接得出\(g(i)=\sum_{i|d}\mu(\frac{d}{n})\lfloor \frac{n}{d} \rfloor \lfloor \frac{m}{d} \rfloor\)

继续推

\[ans'=\sum_{i=1}^{min(n,m)}F(i)\sum_{i|d}\mu(\frac{d}{i})\lfloor \frac{n}{d} \rfloor \lfloor \frac{m}{d} \rfloor=\sum_{i=1}^{min(n,m)}F(i)\sum_{k=1}^{\lfloor \frac{min(n,m)}{i} \rfloor}\mu(k)\lfloor \frac{n}{ik} \rfloor \lfloor \frac{m}{ik} \rfloor
\]

设\(T=ik\)

\[ans'=\sum_{T=1}^{min(n,m)}\lfloor \frac{n}{T} \rfloor \lfloor \frac{m}{T} \rfloor\sum_{k=1}^{\lfloor \frac{min(n,m)}{T} \rfloor}F(k)\mu(\lfloor \frac{T}{k} \rfloor)
\]

前面部分整除分块,我们只要处理出后半部分的前缀和就行了

对于\(F(k)\)直接\(O(nlog_2n)\)暴力枚举约数和它的倍数

对于\(\mu\)这就不用说了吧。。。(看之前的文章)

好的,我们把改装之后的问题解决了,可是原问题呢?

回到真正的\(ans\),因为有a的限制,所以那些\(F(i)\)大于a的是不能加贡献的

那么我们把询问按a排序,然后用树状数组维护\(F(k)\mu(\lfloor \frac{T}{k} \rfloor)\)的前缀和,只有当前询问的a大于扫描到的\(F(i)\),才把\(F(i)\)能产生的贡献加入树状数组。这样就保证了不改加入的贡献不会被加入

#include<bits/stdc++.h>
#define ll long long
const int MAXT=20000+10,MAXN=100000+10,Mod=0x7fffffff;
int T,cnt,vis[MAXN],prime[MAXN],mu[MAXN],C[MAXN],ans[MAXT],limit;
struct question{
int n,m,a;
int id;
inline bool operator < (const question &A) const{
return a<A.a;
};
};
question Q[MAXT];
struct node{
int s;
int id;
inline bool operator < (const node &A) const{
return s<A.s;
};
};
node F[MAXN];
template<typename T> inline void read(T &x)
{
T data=0,w=1;
char ch=0;
while(ch!='-'&&(ch<'0'||ch>'9'))ch=getchar();
if(ch=='-')w=-1,ch=getchar();
while(ch>='0'&&ch<='9')data=((T)data<<3)+((T)data<<1)+(ch^'0'),ch=getchar();
x=data*w;
}
template<typename T> inline void write(T x,char c='\0')
{
if(x<0)putchar('-'),x=-x;
if(x>9)write(x/10);
putchar(x%10+'0');
if(c!='\0')putchar(c);
}
template<typename T> inline void chkmin(T &x,T y){x=(y<x?y:x);}
template<typename T> inline void chkmax(T &x,T y){x=(y>x?y:x);}
template<typename T> inline T min(T x,T y){return x<y?x:y;}
template<typename T> inline T max(T x,T y){return x>y?x:y;}
inline void init()
{
for(register int i=1;i<=limit;++i)
for(register int j=i;j<=limit;j+=i)F[j].s+=i;
for(register int i=1;i<=limit;++i)F[i].id=i;
std::sort(F+1,F+limit+1);
memset(vis,1,sizeof(vis));
vis[0]=vis[1]=0;
mu[1]=1;
for(register int i=2;i<=limit;++i)
{
if(vis[i])
{
prime[++cnt]=i;
mu[i]=-1;
}
for(register int j=1;j<=cnt&&i*prime[j]<=limit;++j)
{
vis[i*prime[j]]=0;
if(i%prime[j])mu[i*prime[j]]=-mu[i];
else break;
}
}
}
inline int lowbit(int x)
{
return x&(-x);
}
inline int sum(int x)
{
int res=0;
while(x)
{
res+=C[x];
x-=lowbit(x);
}
return res;
}
inline void add(int x,int k)
{
while(x<=limit)
{
C[x]+=k;
x+=lowbit(x);
}
}
inline int solve(int n,int m)
{
int res=0;
for(register int i=1;;)
{
if(i>min(n,m))break;
int j=min(n/(n/i),m/(m/i));
res+=(n/i)*(m/i)*(sum(j)-sum(i-1));
i=j+1;
}
return res&Mod;
}
int main()
{
read(T);
for(register int i=1;i<=T;++i)
{
read(Q[i].n);read(Q[i].m);read(Q[i].a);
Q[i].id=i;
chkmax(limit,min(Q[i].n,Q[i].m));
}
init();
std::sort(Q+1,Q+T+1);
for(register int i=1,j=1;i<=T;++i)
{
while(j<=limit&&F[j].s<=Q[i].a)
{
for(register int p=F[j].id,k=1;p<=limit;p+=F[j].id,++k)add(p,F[j].s*mu[k]);
++j;
}
ans[Q[i].id]=solve(Q[i].n,Q[i].m);
}
for(register int i=1;i<=T;++i)write(ans[i],'\n');
return 0;
}