题意:
有n个员工,第k个员工今年可以得到 k4的薪水,
今年老板要解雇所有编号与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;
}