HDU2204Eddy's爱好 数论，容斥原理

HDU原题

题意

给你一个正整数N，确定在1到N之间有多少个可以表示成M^K（K>1) 的数。本题有多组测试数据，每组包含一个整数N，1<=N<=1000000000000000000(10^18).

分析

可以从M,K入手遍历.如果从M遍历再算K次方值，由于数据过大会超时。所以，从K遍历。
K = 2时， M^2， 在[1,N]之内的M数有 floor（ N^(1/2) ）-1个
K =3为 floor（ N^(1/3) ）-1个… …
则 K = k，有 floor（ N^(1/k) ）-1 个。
注意M^6 与（M^2) ^ 3 和(M^3) ^ 2等价，即M^ 6 = M^2 &&M^3 。算到M^6 时，其元素与M^2 与M^3 元素重合，已经被算了两遍，所以根据容斥原理，需要减去M^6 ，即 ans = SUM(奇数个）-SUM（偶数个）。

WA原因

1. 素数筛从2开始
2. 使用for循环确定范围，小心手滑在for循环内重新定义变量
3. 遍历k的取值范围与n有关
4. 开根号有可能小于1 floor（ N^(1/2) ）-1有可能为负数
5. 每写一块功能代码，就想几个样例调试一下是否有效是个好习惯

AC代码

#include <bits/stdc++.h>
using namespace std;
#define N 70
long long  prime[N], check[N], tot;

void eular(){
    memset(check, 0, sizeof(check));
    tot = 0;
    for(long long i = 2; i<N; i++){
        if(!check[i]) prime[tot++] = i;
        for(long long j = 0; j<tot&&prime[j]*i<N; j++){
            check[i*prime[j]] = 1;
            if(i%prime[j] == 0) break;
        }
    }
}

int main(int argc, char const *argv[])
{
    eular();
    
    long long n;
    while( (scanf("%lld", &n)) != EOF){
        long long ret = 0;
        int cntp;//!!
        for(cntp = 0;cntp < tot;cntp++){
            if(pow(2,prime[cntp])>n) break;
        }
        for(long long  i = 1; i<(1LL<<cntp) ;i++){
            int cnt = 0, tmp =1;
            for(long long j = 0; j<cntp; j++){
                if(i&(1LL<<j)){
                    tmp *= prime[j];
                    cnt++;
                }
            }
            double sq = 1.0/tmp;
            long long tmpr = (long long)pow(n, sq) - 1;
            if(tmpr < 0) continue;
            if(cnt&1) ret += tmpr;
            else ret -= tmpr;
        }
        printf("%lld\n", ret+1);
        
    }

    
    return 0;
}