BZOJ2190 洛谷2158 [SDOI2008 仪仗队]

题目描述

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

输入输出格式

输入格式：
共一个数N

输出格式：
共一个数，即C君应看到的学生人数。

输入输出样例

输入样例#1：
4
输出样例#1：
9
说明

【数据规模和约定】

对于 100% 的数据，1 ≤ N ≤ 40000

#include<iostream>
#include<cstdio>
#include<cstring>
#define N 40005
using namespace std;
int prime[N];
bool not_prime[N];
int n,ans,tot=1;
void pre() {
    prime[1]=2;
    not_prime[1]=true;
    for(int i=2; i<=N; i++) {
        if(!not_prime[i]) prime[++tot]=i;
        for(int j=1; j<=tot && i*prime[j]<=N; j++) {
            not_prime[i*prime[j]]=true;
            if(i%prime[j]) break;
        }
    }
}

int get_phi(int x) {
    int ret=1;
    for(int i=1; prime[i]*prime[i]<=x; i++) {
        if(x%prime[i]==0) {
            ret*=prime[i]-1;
            x/=prime[i];
            while(x%prime[i]==0) x/=prime[i],ret*=prime[i];
        }
    }
    if(x>1) ret*=x-1;
    return ret;
}

int main() {
    scanf("%d",&n);
    pre();
    for(int i=1; i<n; i++) ans+=get_phi(i);
    ans*=2;
    ans+=1;
    printf("%d\n",ans);
    return 0;
}