容斥定理

題目

count_prime

Description
給定你一個數n，請你統計出在[a,b]這個區間中和n互質的數的個數。
兩個數互質當且僅當他們除了1之外沒有其他的公共因子或者他們最大的公共因子是1。1和任何數是互素的。
Input
第一行輸入一個整數T（1 <= T <= 100），表示T組測試數據。
接下來T行，每行3個整數a，b，n（1 <= a <=b <=10^15, 1<= n <= 10^9），用空格隔開。
Output
輸出一個整數表示和n互質的數的個數。
Sample Input
2
1 10 2
3 10 5
Sample Output
5
6

思路

互素就是不存在公因子，就是x,y不存在2公因子，3公因子，5公因子。然後用下容斥。

代碼

#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
using namespace std;

int T;
typedef long long ll;
ll a,b,n;
vector<ll> vec;
ll solve(ll b,ll n) {
    vec.clear();
    ll i,j;
    for (i = 2;i * i <= n;i ++) {
        if(n % i == 0) {
            vec.push_back(i);
            while(n % i == 0) n /= i;
        }
    }
    if(n > 1) vec.push_back(n);
    int len = vec.size();
    ll tot = 0;
    for (i = 1;i < (1 << len);i ++) {
        ll cnt = 0;
        ll sum = 1;
        for (j = 0;j < len;j ++) {
            if(i >> j & 1) {
                cnt ++;
                sum *= vec[j];
            }
        }
        if(cnt & 1) {
            tot += b / sum;
        }
        else {
            tot -= b / sum;
        }
    }
    return b - tot;
}

int main()  {
    scanf("%d",&T);
    while(T --) {
        scanf("%lld%lld%lld",&a,&b,&n);
        printf("%lld\n",solve(b,n) - solve(a - 1,n));
    }
}