problem
有一张 n × m n\times m n×m 的数表,其第 i i i 行第 j j j 列的数值为能同时整除 i i i 和 j j j 的所有自然数之和。
给定 a a a,计算数表中 ≤ a \le a ≤a 的数之和。
数据范围: 1 ≤ n , m ≤ 1 0 5 1 \le n,m \le 10^5 1≤n,m≤105, 1 ≤ T ≤ 2 × 1 0 4 1 \le T \le2\times10^4 1≤T≤2×104, a ≤ 1 0 9 a\le10^9 a≤109,其中 T T T 是数据组数。
solution
设 σ ( n ) \sigma(n) σ(n) 表示 n n n 的所有约数之和,即 σ ( n ) = ∑ d ∣ n d \sigma(n)=\sum_{d|n}d σ(n)=∑d∣nd。
我们先不考虑 a a a 的限制来推一下式子:
a n s = ∑ i = 1 n ∑ j = 1 m σ ( gcd ( i , j ) ) ans=\sum_{i=1}^n\sum_{j=1}^m\sigma(\gcd(i,j)) ans=i=1∑nj=1∑mσ(gcd(i,j))
由于 a ∣ i , a ∣ j ⟹ a ∣ gcd ( i , j ) a\mid i,a\mid j\Longrightarrow a\mid\gcd(i,j) a∣i,a∣j⟹a∣gcd(i,j),所以后面那个直接是 σ ( gcd ( i , j ) ) \sigma(\gcd(i,j)) σ(gcd(i,j))。
那么枚举 gcd \gcd gcd(设 u p = min ( n , m ) up=\min(n,m) up=min(n,m)):
a n s = ∑ k = 1 u p σ ( k ) ∑ i = 1 ⌊ n k ⌋ ∑ j = 1 ⌊ m k ⌋ [ gcd ( i , j ) = 1 ] ans=\sum_{k=1}^{up}\sigma(k)\sum_{i=1}^{\lfloor\frac n k\rfloor}\sum_{j=1}^{\lfloor\frac m k\rfloor}[\gcd(i,j)=1] ans=k=1∑upσ(k)i=1∑⌊kn⌋j=1∑⌊km⌋[gcd(i,j)=1]
然后就是莫反:
a n s = ∑ k = 1 u p σ ( k ) ∑ i = 1 ⌊ n k ⌋ ∑ j = 1 ⌊ m k ⌋ ∑ d ∣ i , d ∣ j μ ( d ) = ∑ k = 1 u p σ ( k ) ∑ d = 1 ⌊ u p k ⌋ μ ( d ) ∑ i = 1 ⌊ n k d ⌋ ∑ j = 1 ⌊ m k d ⌋ 1 = ∑ k = 1 u p σ ( k ) ∑ d = 1 ⌊ u p k ⌋ μ ( d ) ⌊ n k d ⌋ ⌊ m k d ⌋ \begin{aligned} ans&=\sum_{k=1}^{up}\sigma(k)\sum_{i=1}^{\lfloor\frac n k\rfloor}\sum_{j=1}^{\lfloor\frac m k\rfloor}\sum_{d|i,d|j}\mu(d)\\ &=\sum_{k=1}^{up}\sigma(k)\sum_{d=1}^{\lfloor\frac {up} k\rfloor}\mu(d)\sum_{i=1}^{\lfloor\frac n {kd}\rfloor}\sum_{j=1}^{\lfloor\frac m {kd}\rfloor}1\\ &=\sum_{k=1}^{up}\sigma(k)\sum_{d=1}^{\lfloor\frac {up} k\rfloor}\mu(d)\lfloor\frac n {kd}\rfloor\lfloor\frac m {kd}\rfloor \end{aligned} ans=k=1∑upσ(k)i=1∑⌊kn⌋j=1∑⌊km⌋d∣i,d∣j∑μ(d)=k=1∑upσ(k)d=1∑⌊kup⌋μ(d)i=1∑⌊kdn⌋j=1∑⌊kdm⌋1=k=1∑upσ(k)d=1∑⌊kup⌋μ(d)⌊kdn⌋⌊kdm⌋
设 T = k d T=kd T=kd,那么有:
a n s = ∑ T = 1 u p ∑ k ∣ T σ ( k ) μ ( T k ) ⌊ n T ⌋ ⌊ m T ⌋ ans=\sum_{T=1}^{up}\sum_{k|T}\sigma(k)\mu(\frac T k)\lfloor\frac n {T}\rfloor\lfloor\frac m {T}\rfloor ans=T=1∑upk∣T∑σ(k)μ(kT)⌊Tn⌋⌊Tm⌋
所以如果没有 a a a 的限制,这道题直接就做完了。
我们设 f ( n ) = ∑ k ∣ n σ ( k ) μ ( n k ) f(n)=\sum_{k|n}\sigma(k)\mu(\frac n k) f(n)=∑k∣nσ(k)μ(kn),那么只有 σ ( k ) ≤ a \sigma(k)\le a σ(k)≤a 的部分才会对答案产生贡献。
把询问离线下来,对 a a a 排序,然后把 σ \sigma σ 的函数值也记下来排序,那么就用双指针,当进行下一个询问时,就先把 ≤ a \le a ≤a 的所有 σ \sigma σ 值全部加进 f f f 函数里(就是枚举倍数加就可以了)再处理。
发现维护 f f f 时,需要单点加,区间查值(在整除分块的时候),因此我们用树状数组维护即可。
时间复杂度 O ( T n log n + n log 2 n ) O(T\sqrt n \log n+n\log^2n) O(Tnlogn+nlog2n)。
PS:由于要对 2 31 2^{31} 231 取模,我们可以先 unsigned int 自然溢出(对 2 32 2^{32} 232 取模),最后再对 2 31 2^{31} 231 取模。
code
#include<bits/stdc++.h>
#define N 100005
#define uint unsigned int
#define lowbit(x) (x&-x)
using namespace std;
struct Query{int n,m,a,id;}Q[N],S[N];
uint mu[N],sigma[N],ans[N],bit[N];
bool operator<(const Query &p,const Query &q) {return p.a<q.a;}
void add(int x,uint val) {for(;x<N;x+=lowbit(x))bit[x]+=val;}
uint Query(int x,uint ans=0) {for(;x;x-=lowbit(x))ans+=bit[x];return ans;}
void modify(int x) {for(int i=1;i*x<N;++i)add(i*x,sigma[x]*mu[i]);}
int sum,prime[N],mark[N];
void linear_sieves(){
mu[1]=1;
for(int i=2;i<N;++i){
if(!mark[i]) prime[++sum]=i,mu[i]=-1;
for(int j=1;j<=sum&&i*prime[j]<N;++j){
mark[i*prime[j]]=1;
if(i%prime[j]) mu[i*prime[j]]=-mu[i];
else {mu[i*prime[j]]=0;break;}
}
}
for(int i=1;i<N;++i)
for(int j=1;i*j<N;++j) sigma[i*j]+=i;
for(int i=1;i<N;++i) S[i].id=i,S[i].a=sigma[i];
sort(S+1,S+N);
}
uint solve(int n,int m){
if(n>m) swap(n,m);
uint ans=0;
for(int i=1,j;i<=n;i=j+1){
j=min(n/(n/i),m/(m/i));
ans+=(n/i)*(m/i)*(Query(j)-Query(i-1));
}
return ans;
}
int main(){
int T;
scanf("%d",&T);
for(int i=1;i<=T;++i){
scanf("%d%d%d",&Q[i].n,&Q[i].m,&Q[i].a),Q[i].id=i;
}
sort(Q+1,Q+T+1),linear_sieves();
int now=0;
for(int i=1;i<=T;++i){
int n=Q[i].n,m=Q[i].m,a=Q[i].a;
while(now<N-1&&S[now+1].a<=a) modify(S[++now].id);
ans[Q[i].id]=solve(n,m);
}
uint P=1ll<<31;
for(int i=1;i<=T;++i) printf("%d\n",ans[i]%P);
return 0;
}