华为面试题：完全数定义与欧拉公式求解完全数

题目

完全数（Perfect number），又称完美数或完备数，是一些特殊的自然数。
它所有的真因子（即除了自身以外的约数）的和（即因子函数），恰好等于它本身。
例如：28，它有约数1、2、4、7、14、28，除去它本身28外，其余5个数相加，1+2+4+7+14=28。
输入n，请输出n以内(含n)完全数的个数。

数据范围： 1≤n≤5×10^5
输入描述：
输入一个数字n
输出描述：
输出不超过n的完全数的个数

示例1
输入：

1000

输出：

3

完全数定义与计算（欧拉）

大数学家欧拉曾推算出完全数的获得公式：

如果p是质数，且2^p-1 也是质数，那么（2^p-1）X 2^（p-1）便是一个完全数。

例如p=2，是一个质数，2^{p-1=3也是质数，（2}p-1）X2^（p-1）=3X2=6，是完全数。
例如p=3，是一个质数，2^{p-1=7也是质数，（2}p-1）X2^（p-1）=7X4=28，是完全数。
例如p=5，是一个质数，2^{p-1=31也是质数，（2}p-1）X2^（p-1）=31X16=496是完全数。

当2^p-1是质数的时候，称其为梅森素数。到2013年2月6日为止，人类只发现了48个梅森素数，较小的有3、7、31、127等。

代码

#include<iostream>
#include<vector>

using namespace std;


//线性筛计算素数
vector<bool> is_primer; //判断下标i是否为素数
vector<int> primer; //素数数组
int primer_cnt;  //素数个数
void func(int n){
    is_primer = vector<bool>(n+1,true);
    for(int i = 2; i <= n; ++i){
        if(is_primer[i]){
            primer_cnt++;
            primer.push_back(i);
        }
        for(int j = 0; j <= primer_cnt && primer[j]*i <= n; ++j){
            is_primer[i * primer[j]] = false;
            if(i % primer[j] == 0) break;
        }
    }
    return ;
}

//欧拉计算完全数公式：
//如果p是质数，且2^p-1 也是质数，那么（2^p-1）X 2^（p-1）便是一个完全数。
int cal_ab_num(int n){
    int cnt = 0;
    func(n);
    for(int i = 0; i < primer_cnt;++i){
        int a = primer[i];
        int b = 1;
        for(int j = 0; j < a; ++j){
            b = b << 1;
            if(b > n) return cnt;
        }
        b = b - 1;
        int tmp = 1;
        tmp = tmp << (a - 1);
        if( b * tmp > n) return cnt;
        if(is_primer[b]) {
            cnt++;
            //cout << a << " " << b << " val = " << b * tmp << endl;
        }
    }
    return cnt;
}




int main(){
    int num;
    cin >> num;
    cout << cal_ab_num(num) << endl;
    return 0;
}