bzoj 3529 数表

题目需要求出$\sum_{1\le i \le n}\sum_{1\le j \le m}\sigma((i,j))|\sigma((i,j))\le a$。考虑处理$[1,10^5]$的所有数的因数和，排序之后，查询会在一个前缀上完成。即对于一个查询，应求的是：$\sum_{1\le d \le 10^5}F(d)\sum_{x|d \land \sigma(x)\le a}\mu(\frac d x)\sigma(x)$。那么按$a$升序离线处理每个查询，用树状数组或者线段树动态更新区间状态即可。因为一个数只会更新一次，故总的更新复杂度为$O(nlog^2n)$，总查询复杂度为$O(n\sqrt n logn)$。

#include<algorithm>
#include<iostream>
#include<cstring>
#include<climits>
#include<cstdio>
//--container
using namespace std;
typedef long long ll;
typedef pair<int,int> pi;
const ll md=(ll)1<<31;
const int up=100000;
//--
ll te[100010],rho[100010];int q,mu[100010],fn=0,fr[100010],sr[100010];
#define clr(a) memset(a,0,sizeof(a))
inline int lw(int x){return x&-x;};
void upd(int d,ll x){
    for(;d<=up;d+=lw(d))te[d]=(te[d]+x)%md;
};
ll _qry(int d){
    ll r=0;for(;d;d-=lw(d))r=(r+te[d])%md;
    return r;
};
inline ll qry(int l,int r){return (_qry(r)-_qry(l-1)+md)%md;};
bool smp(const int&a,const int&b){return rho[a]<rho[b];};
void clit(){
    int i,j,t;for(i=1;i<=up;mu[i++]=-2);
    for(mu[1]=1,i=2;i<=up;++i){
        if(mu[i]==-2)mu[i]=-1,fr[fn++]=i;
        for(j=0;j<fn&&fr[j]*i<=up;++j){
            if(!(i%fr[j])){mu[fr[j]*i]=0;continue;}
            mu[fr[j]*i]=-mu[i];
        }
    }
    for(i=1;i<=up;++i)for(j=i;j<=up;j+=i)rho[j]+=i;
    for(i=1;i<=up;++i)sr[i]=i;sort(sr+1,sr+up+1,smp);
};
struct qr{int n,m,a,i;}qsr[20010];
bool qrmp(const qr&a,const qr&b){return a.a<b.a;};
bool _qrmp(const qr&a,const qr&b){return a.i<b.i;};
ll _cl(int n,int m){
    ll r=0;int i,j;if(n>m)swap(n,m);
    for(i=1;i<=n;i=j+1){
        j=min(n/(n/i),m/(m/i));
        r=(r+qry(i,j)*(ll)(n/i)*(ll)(m/i))%md;
    }
    return r;
};
void __cl(int d){
    int i;ll t;for(i=d;i<=up;i+=d){
        t=(rho[d]*mu[i/d]+md)%md;upd(i,t);
    }
};
void cl(){
    int i,j,t,a,b;scanf("%d",&q);
    for(i=0;i<q;++i){
        scanf("%d %d %d",&qsr[i].n,&qsr[i].m,&qsr[i].a);qsr[i].i=i;
    }
    for(sort(qsr,qsr+q,qrmp),i=0,j=1;i<q;++i){
        for(;j<=up&&rho[sr[j]]<=qsr[i].a;__cl(sr[j++]));
        qsr[i].a=_cl(qsr[i].n,qsr[i].m);
    }
    for(sort(qsr,qsr+q,_qrmp),i=0;i<q;printf("%d\n",qsr[i++].a));
};
int main(){
#ifndef ONLINE_JUDGE
    freopen("in.txt","r",stdin);
    freopen("out.txt","w",stdout);
#endif
    clit();cl();
    return 0;
};