hdu 5212 code

题目传送门

法一：观察calc函数，其本质就是求sigma(x*(x-1))，x=gcd(a[i],a[j])。然后发现a[i]的范围不大，因此x的范围也不大。

于是可以考虑数据范围内每个x对答案的贡献。枚举x，设f[i]表示gcd=i的数对个数。则累加f[i]*i*(i-1)即为结果。

关键就是求f[i]。

可以先计算出n个数中i的倍数有多少个，设为s。这s个数，任意两个数的gcd都是i的倍数且gcd=i的两个数必然从这s个数中选出，于是总的数对个数为s*s，减去gcd不是i的个数（即gcd=2i,3i……）便可求得f[i]。即f[i]=s*s-f[2i]-f[3i]-……

时间复杂度为O(nlogn)

#include<stdio.h>
#include<string.h>
long long f[10005],an;
int a[10005];
int main()
{
    int i,s,n,k,j;
    while(~scanf("%d",&n))
    {
        memset(a,0,sizeof(a));
        an=0;
        for(i=1;i<=n;++i) {scanf("%d",&k);a[k]++;}
        for(i=10000;i>=1;--i)
        {
            s=0;
            for(j=i;j<=10000;j+=i) {s+=a[j];f[i]-=f[j];}
            f[i]+=s*s;
            an=(an+f[i]*i*(i-1))%10007;
        }
        printf("%d\n",an);
    }
    return 0;
}

法二：莫比乌斯反演

官方题解：

定义f(x)=x∗(x−1)
题目所求即为sigma(f(gcd(ai,aj)|i!=j,1≤i,j≤n)
先用线性筛求出miu在[1,10000]的函数值
利用莫比乌斯反演公式我们可以O(vlogv)暴力求解出函数g在[1,10000]的函数值，其中g满足：
sigma(g(d)|x
这样所求答案即为：
sigma(g(d)∗cnt(d)∗cnt(d)|1≤d≤10000)，其中cnt函数满足：
cnt(x)=在a1,a2,..,an中是x的倍数的个数
而cnt的取值也可以O(vlogv)暴力计算出
所以总的时间复杂度就是O(vlogv)的

设f[i]表示gcd=i的数对个数

F[i]表示gcd=i的倍数的数对个数。F[i]易求。然后莫比乌斯反演求f[i]，结果累加f[i]*i*(i-1)即为答案。

#include<stdio.h>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 10005
typedef long long LL;
int u[N],a[N],p[N],t;
LL f,F[N];
bool v[N];
void init()
{
    int i,j,t=0;
    u[1]=1;
    for(i=2;i<N;++i)
    {
        if(!v[i]) p[t++]=i,u[i]=-1;
        for(j=0;j<t&&p[j]*i<N;++j)
        {
            v[p[j]*i]=1;
            if(i%p[j]==0) {u[p[j]*i]=0;break;}
            u[i*p[j]]=-u[i];
        }
    }
}

int main()
{
    int i,j,k,n,M;
    LL ans;
    init();
    while(~scanf("%d",&n))
    {
        ans=M=0;
        memset(a,0,sizeof(a));
        for(i=1;i<=n;++i) {scanf("%d",&k);a[k]++;M=max(M,k);}
        for(i=1;i<=M;++i)
        {
            for(F[i]=0,j=i;j<=M;j+=i) F[i]=F[i]+a[j];
            F[i]*=F[i];
        }
        for(i=1;i<=M;++i)
        {
            for(f=0,j=i;j<=M;j+=i) f+=u[j/i]*F[j];
            ans=(ans+f*(i-1)*i%10007)%10007;
        }
        printf("%lld\n",ans);
    }
    return 0;
}