P2158题解

发布时间 2023-08-14 16:12:00作者: DDream白日梦

P2158题解

[SDOI2008] 仪仗队

题目描述

作为体育委员,C 君负责这次运动会仪仗队的训练。仪仗队是由学生组成的 $N \times N$ 的方阵,为了保证队伍在行进中整齐划一,C 君会跟在仪仗队的左后方,根据其视线所及的学生人数来判断队伍是否整齐(如下图)。

现在,C 君希望你告诉他队伍整齐时能看到的学生人数。

题解

首先考虑从C君引出的一条直线上只会有一个点产生贡献,由此想到一次函数。

考虑以C君为原点建系,显然只有横纵坐标互质的点会产生贡献,又因为横纵坐标对称,所以只需要算出1n-1的欧拉函数之和乘2即可,特判(1,1)(注意这里是1n-1的和,部分题解给出的解释是因为算的行列是1~n-1,但其实只是碰巧相等而已,实质是多出的,和少的部分相等)。

最后注意一下欧拉函数和埃氏筛的求法。

代码:

#include<bits/stdc++.h>
using namespace std;
const int N=4e4+100;
int n,p[N],ans;
int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0),cout.tie(0);
	cin>>n;
	for(int i=1;i<=n;i++)
	p[i]=i;
	for(int i=2;i<=n;i++)
	if(p[i]==i)
	for(int j=i;j<=n;j+=i)
	p[j]-=p[j]/i;
	for(int i=1;i<=n-1;i++)
	ans+=p[i]; 
	ans*=2;
	if(n!=1)
	ans+=1;
	cout<<ans;
	return 0;
}