Description
作为体育委员,C君负责这次运动会仪仗队的训练。仪仗队是由学生组成的N * N的方阵,为了保证队伍在行进中整齐划一,C君会跟在仪仗队的左后方,根据其视线所及的学生人数来判断队伍是否整齐(如下图)。
现在,C君希望你告诉他队伍整齐时能看到的学生人数。
Input
共一个数N。
Output
共一个数,即C君应看到的学生人数。
Sample Input
4
Sample Output
9
HINT
【数据规模和约定】 对于 100% 的数据,1 ≤ N ≤ 40000
Solution
以前看这题,无从下手
现在看这题,毫无意义
由于这个什么人在队列的左下角,所以我们先把第一列和最后一行去掉,剩下的变成一个矩阵
然后对于这个矩阵
\(ans'=\sum_{i=1}^n\sum_{j=1}^n[gcd(i,j)=1]\)
\(\ \ \ \ \ \ \ \ =\sum_{i=1}^n\mu(i)(\lfloor \frac{n}{i}\rfloor )^2\)
然后加上原来的最后一行和第一列的贡献,就是加2
答案就是\(ans=ans'+2\)
当然,这种做法要特判一下,因为我们剔掉了一行一列,那么当 \(n=1\) 的时候,答案特判,为0
#include<bits/stdc++.h>
#define ll long long
#define db double
#define ld long double
const int MAXN=40000+10;
int n,vis[MAXN],prime[MAXN],cnt,s[MAXN],mu[MAXN];
ll res=0;
template<typename T> inline void read(T &x)
{
T data=0,w=1;
char ch=0;
while(ch!='-'&&(ch<'0'||ch>'9'))ch=getchar();
if(ch=='-')w=-1,ch=getchar();
while(ch>='0'&&ch<='9')data=((T)data<<3)+((T)data<<1)+(ch^'0'),ch=getchar();
x=data*w;
}
template<typename T> inline void write(T x,char c='\0')
{
if(x<0)putchar('-'),x=-x;
if(x>9)write(x/10);
putchar(x%10+'0');
if(c!='\0')putchar(c);
}
template<typename T> inline void chkmin(T &x,T y){x=(y<x?y:x);}
template<typename T> inline void chkmax(T &x,T y){x=(y>x?y:x);}
template<typename T> inline T min(T x,T y){return x<y?x:y;}
template<typename T> inline T max(T x,T y){return x>y?x:y;}
inline void init()
{
memset(vis,1,sizeof(vis));
vis[0]=vis[1]=1;
mu[1]=1;
for(register int i=2;i<MAXN;++i)
{
if(vis[i])
{
prime[++cnt]=i;
mu[i]=-1;
}
for(register int j=1;j<=cnt&&i*prime[j]<MAXN;++j)
{
vis[i*prime[j]]=0;
if(i%prime[j])mu[i*prime[j]]=-mu[i];
else break;
}
}
for(register int i=1;i<MAXN;++i)s[i]=s[i-1]+mu[i];
}
int main()
{
init();
read(n);
if(n<=1)
{
puts("0");
return 0;
}
n--;
for(register int i=1;;)
{
if(i>n)break;
int j=n/(n/i);
res+=1ll*(n/i)*(n/i)*(s[j]-s[i-1]);
i=j+1;
}
write(res+2,'\n');
return 0;
}