HDU 5663 Hillan and the girl (莫比乌斯反演+组合性质)好题

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=5663

#include<bits/stdc++.h>
using namespace std;

#define debug puts("YES");
#define rep(x,y,z) for(int (x)=(y);(x)<(z);(x)++)
#define read(x,y) scanf("%d%d",&x,&y)
#define ll long long

#define lrt int l,int r,int rt
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define root l,r,rt
const int  maxn =1e7+5;
const int mod=1e9+7;
const int ub=1e6;
ll powmod(ll x,ll y){ll t; for(t=1;y;y>>=1,x=x*x%mod) if(y&1) t=t*x%mod; return t;}
ll gcd(ll x,ll y){return y?gcd(y,x%y):x;}
/*
题目大意：求式子，二维前缀和形式，里面的权值是f(gcd(i,j))，
其中f函数是如果x是平方数则为0，不是则为1.

太老套的形式，看到想哭，直接开撸式子吧，
最终经典套路求出来的是：
sigma T=1~n n/T*m/T* (miu*f)(T)。
题目明显要求O(n^(1/2))的复杂度，分块等技巧都是老套，
主要是后面的miu*f如何处理，
我们分析下，f函数规定只有当x不是完全平方数时才有权值1，
miu(d)*f(T/d)的sigma形式是我们的研究目标，
因为miu的限制，每个质因子至多只能取一个，
有没有灵感了？取一个其质因子幂数就改变了不是嘛，不取不就是不改变嘛，
组合性质就出来了，把T的质因子分类成两部分，分奇偶，
奇数的个数为a,偶数的个数为b,这里我们反面考虑一下，
因为如果f函数为恒一函数，则研究目标值为0，（基本性质），
那么我们不妨用0减去，f函数为0时产生的权值，
很显然只有一种情况，f函数为零，d 的组合选择恰好选择了奇数幂次质数组合，
就是说最终F(T)=-(-1)^(T中奇数次幂质因子个数),
明显积性函数欧拉筛啊~等等！
这样还不完美，再思考下，其实F(T)就是(-1)^(质因子幂次和)，
因为偶数次幂产生的影响是无效的，这样欧拉筛时就更简单了。一次AC

*/
///筛法筛莫比乌斯反演函数素数等
int prim[maxn/10],tot;
bool vis[maxn];
short miu[maxn];
int f[maxn];
void sieve()
{
    memset(vis,false,sizeof(vis));
    miu[1]=1,f[1]=0;
    for(int i=2;i<maxn;i++)
    {
        if(!vis[i]) prim[tot++]=i,miu[i]=-1,f[i]=-1;
        for(int j=0;j<tot;j++)
        {
            if(1LL*i*prim[j]>=maxn) break;
            int k=i*prim[j];
            vis[k]=1; f[k]=-f[i];
            if(i%prim[j])  miu[k]=-miu[i];
            else  break;
        }
    }
    for(int i=1;i<maxn;i++)  f[i]=f[i-1]-f[i];
}

ll n,m;
int main()
{
    sieve();
    int t;scanf("%d",&t);
    while(t--)
    {
        scanf("%lld%lld",&n,&m);
        if(n>m) swap(n,m);
        ll ans=0;
        for(ll i=1,j;i<=n;i=j+1)
        {
            j=min(n/(n/i),m/(m/i));
            ans+=(n/i)*(m/i)*(f[j]-f[i-1]);
        }
        printf("%lld\n",ans);
    }
    return 0;
}