hdu——3501（数论之容斥原理）

#include <iostream>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <cstdio>
using namespace std;
typedef long long ll;
#define INF 0xfffffff
#define MAX(a,b) a>b?a:b
#define MIN(a,b) a>b?b:a

#define MOD 1000000007
#define M 1000000
short int flag[M];
int prime[200000];//当M比较大的时候(大于10e5)素数大概就是范围的1/10
int q=0;//素数的下标
void sieve_prime()                            //筛法求素数
{
memset(flag,0,sizeof(flag));
int i;
for(i=2;i*i<M;i++) {    //边筛边存
if(flag[i]) continue;
prime[q++]=i;
for(int j=i*i;j<M;j+=i)
flag[j]=1;
}
for(i;i<M;i++)   //存另一部分素数
if(!flag[i]) prime[q++]=i;
}
ll a[30];
int p;
void resolve(int n)  //分解N
{
p=0;
for(int i=0;prime[i]*prime[i]<=n&&n!=1;i++)
{
if(n%prime[i]==0)
{
a[p++]=(ll)prime[i];
while(n%prime[i]==0) n/=prime[i];
}
}
if(n!=1) a[p++]=n;
}
void deal_with(ll n)
{
ll ans=0;
int end=1<<p;
int temp;
int k;
int count_1;
ll mid;
ll item;
ll item_sum;
for(int i=1;i<end;i++)
{
temp=i;
k=0;
count_1=0;
mid=1;
while(temp)
{
if(temp&1) count_1++,mid*=a[k];  //mid记录的是选取的几个素数（1的时候选取）的乘积
k++;   //k表示选取的那个数。
temp/=2;
}
item=((n-1))/mid;
//下面为等差数列求和
item_sum=((item*mid)%MOD+((((item)*(item-1)/2)%MOD)*mid)%MOD)%MOD;//处理仔细点
if(count_1%2) ans=(ans+item_sum)%MOD;  //count_1记录的是选取的个数
else ans=(MOD-item_sum+ans)%MOD;
}
cout<<ans<<endl;
}
int main()
{
int i,j,k,t;
ll m,n;
sieve_prime() ;
while(cin>>n&&n)
{
resolve(n);
deal_with(n);
}
return 0;
}