hdu4059 The Boss on Mars（容斥+自然数前n项4次方和）

题意：

有n个员工，第k个员工今年可以得到 k⁴的薪水，
今年老板要解雇所有编号与n互质的员工，问能省下来多少钱。
答案对1e9+7取模。

数据范围：n<=1e8

解法：

找互质的很麻烦,考虑计算不互质的,
不互质即存在公共质因子,
先计算出n的所有质因子,
根据容斥原理：
加上至少一个公共质因子的倍数的数的4次方和,
减去至少两个公共质因子的倍数的数的4次方和,
加上至少三个公共质因子的倍数的数的4次方和,
...

二进制枚举质数的选择方案,奇加偶减即可.

还有一些细节在代码里面注释了

这题还需要用到1到n的4次方和,直接推的话比较恶心,
最好记一下公式.

code：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxm=1e4+5;
const ll mod=1e9+7;
ll notprime[maxm];
ll prime[maxm],cnt;
ll inv[maxm];
vector<ll>fac;
ll n;
ll ppow(ll a,ll b,ll mod){
    ll ans=1%mod;a%=mod;
    while(b){
        if(b&1)ans=ans*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return ans;
}
void init(){
    for(int i=1;i<=30;i++){
        inv[i]=ppow(i,mod-2,mod);
    }
    for(int i=2;i<maxm;i++){
        if(!notprime[i]){
            prime[cnt++]=i;
        }
        for(int j=0;j<cnt;j++){
            if(prime[j]*i>=maxm)break;
            notprime[prime[j]*i]=1;
            if(i%prime[j]==0)break;
        }
    }
}
ll cal(ll x){//1到n的4次方和
    ll ans=0;
    ans+=ppow(x,5,mod)*inv[5]%mod;
    ans+=ppow(x,4,mod)*inv[4]%mod;
    ans+=ppow(x,3,mod)*inv[3]%mod;
    ans-=x*inv[30]%mod;
//    //下面那种也行
//    ll ans=x*(x+1)%mod*(2*x+1)%mod*(3*x*x%mod+3*x%mod-1+mod)%mod*inv[30]%mod;
    return (ans%mod+mod)%mod;
}
ll solve(){
    ll ans=0;
    int m=fac.size();
    for(int i=1;i<(1<<m);i++){
        int cnt=0;
        ll temp=1;
        for(int j=0;j<m;j++){
            if(i>>j&1){
                cnt++;
                temp*=fac[j];
            }
        }
        ll num=n/temp;//[1,n]中temp的倍数有num个
        //一定是temp,temp*2,temp*3的形式
        //贡献为temp^4+(temp^4)*(2^4)...
        //temp^4提出来,变成(temp^4)*(1^4+2^4....num^4)
        ll t1=temp*temp%mod*temp%mod*temp%mod*cal(num)%mod;
        if(cnt%2){
            ans+=t1;
            ans%=mod;
        }else{
            ans-=t1;
            ans%=mod;
        }
    }
    return (ans%mod+mod)%mod;
}
signed main(){
    init();
    ll T;scanf("%lld",&T);
    while(T--){
        scanf("%lld",&n);
        ll x=n;
        fac.clear();
        for(int j=0;j<cnt;j++){
            if(x%prime[j]==0){
                fac.push_back(prime[j]);
                while(x%prime[j]==0){
                    x/=prime[j];
                }
            }
        }
        if(x!=1)fac.push_back(x);
        ll temp=solve();
        ll ans=((cal(n)-temp)%mod+mod)%mod;
        printf("%lld\n",ans);
    }
    return 0;
}

---