bzoj 2190: [SDOI2008]仪仗队

Description

　　作为体育委员，C君负责这次运动会仪仗队的训练。仪仗队是由学生组成的N * N的方阵，为了保证队伍在行进中整齐划一，C君会跟在仪仗队的左后方，根据其视线所及的学生人数来判断队伍是否整齐(如下图)。 　　  　　现在，C君希望你告诉他队伍整齐时能看到的学生人数。

Input

　　共一个数N。

Output

　　共一个数，即C君应看到的学生人数。

Sample Input

　　4

Sample Output

　　9

HINT

【数据规模和约定】 　　对于 100% 的数据，1 ≤ N ≤ 40000

为何大家都喜欢用欧拉函数OTL就我一个人喜欢用莫比乌斯么。。

【下面mu[]为莫比乌斯函数】

ans=mu[i]*[n/i]*[n/i]+2

n读入之后记得-1

#include<cstdio>
#include<string>
#include<cstring>
using namespace std;
long long mu[1000001],prime[1000001];
bool check[1000001]; 
long long n;
int tot;
inline void findmu()
{
     memset(check,false,sizeof(check));
     mu[1]=1;
     int i,j;
     for(i=2;i<=n;i++)
     {
          if(!check[i])
          {
               tot++;
               prime[tot]=i;
               mu[i]=-1;
          }
          for(j=1;j<=tot;j++)
          {
               if(i*prime[j]>n)
                    break;
               check[i*prime[j]]=true;
               if(i%prime[j]==0)
               {
                    mu[i*prime[j]]=0;
                    break;
               }
               else
                    mu[i*prime[j]]=-mu[i];
          }
     }
}
int main()
{
     scanf("%lld",&n);
     n--;
     findmu();
     long long i;
     long long ans=0;
     for(i=1;i<=n;i++)
          ans+=mu[i]*(n/i)*(n/i);
     ans=ans+(long long)2;
     printf("%lld\n",ans);
     return 0;
}