SDOI2014 数表

SDOI2014 数表

有一张 n×m 的数表，其第 i 行第 j 列（1$\leq$i$\leq$N,1$\leq$j$\leq$m, 1 $\leq$ i $\leq$ N, 1 $\leq$ j $\leq$ m,1$\leq$i$\leq$N,1$\leq$j$\leq$m）的数值为能同时整除 i 和 j 的所有自然数之和。

给定 a，计算数表中不大于 a 的数之和。

---

• 注意把有限制的地方看作函数，整体处理，不要分开，布尔表达式可以放到最后考虑
• $d|i,j$ 即 $d|gcd(i,j)$
• 注意离线做法
• 枚举约数 只与约数有关放到前面

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <vector>
#define File(x) "test."#x
#define For(i,s,e) for(int i=(s); i<=(e); i++)
#define Rep(i,s,e) for(int i=(s); i>=(e); i--)
#define lowbit(x) ((-x)&x)
using namespace std;
typedef long long LL;

const int N=200000+1, C=200000+1;
struct Sig{
    int x,s;
}S[N];
struct Query{
    int x,y,a,to;
}a[N];
bool cmp1(Query a, Query b){
    return a.a<b.a;
}
bool cmp2(Sig a, Sig b){
    return a.s<b.s;
}

int prime[N],minprm[N],minv[N],mina[N];
int f[N],u[N],T[N];
int n,m,Q,mxn,ans[N];
bool notp[N];

void Add(int p, int x){
    while(p<=mxn){
        T[p]+=x; p+=lowbit(p); 
    }
}
int query(int p){
    int ans=0;
    if(!p) return 0;
    while(p){
        ans+=T[p]; p-=lowbit(p);
    }
    return ans;
}

void init(){
    u[1]=1; f[1]=1; S[1]=(Sig){1,1};
    For(i,2,N-1){
        if(!notp[i]){
            prime[++prime[0]]=i; minprm[i]=1+i; minv[i]=i; u[i]=-1; f[i]=i+1;
        }
        for(int j=1; j<=prime[0] && 1LL*i*prime[j]<N; j++){
            int t=i*prime[j];
            notp[t]=1;
            if(i%prime[j]==0){
                u[t]=0;
                minv[t]=minv[i]*prime[j];
                minprm[t]=minprm[i]+minv[t];
                f[t]=f[i]/minprm[i]*minprm[t];
                break;
            }
            u[t]=-u[i]; minprm[t]=1+prime[j]; minv[t]=prime[j]; f[t]=f[i]*minprm[t];
        }
        S[i]=(Sig){i,f[i]};
    }
}

int main()
{
    freopen(File(in),"r",stdin);
    freopen(File(out),"w",stdout);

    init();

    scanf("%d",&Q);
    For(i,1,Q){
        scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].a);
        if(a[i].x>a[i].y) swap(a[i].x, a[i].y);
        mxn=max(mxn, a[i].y); a[i].to=i;
    }
    sort(a+1,a+1+Q,cmp1); sort(S+1,S+1+mxn-1,cmp2);
    int p=1;
    For(i,1,Q){

        while(p<=mxn && S[p].s<=a[i].a){
            for(int d=S[p].x; d<=mxn; d+=S[p].x){
                Add(d,S[p].s*u[d/S[p].x]);
            }
            p++;
        }
        int last=1; n=a[i].x; m=a[i].y;
        for(int d=1; d<=n; d=last+1){
            last=min(n/(n/d), m/(m/d));
            ans[a[i].to]+=(query(last)-query(d-1))*(n/d)*(m/d);
        }
    }
    For(i,1,Q){
        if(ans[i]<0) ans[i]+=(1<<31);
        printf("%d\n",ans[i]);
    }

    return 0;
}