Codeforces Round #589 (Div. 2) C. Primes and Multiplication (思维 + 容斥)

链接：https://codeforces.com/contest/1228/problem/C
来源：Codeforces Round #589 (Div. 2)

  题意：给出一些定义，对于一个 x ，prime(x) ={p₁, p₂…, p_n}，g(x, p) 表示 x 最大可以整除 p 的多少次幂，这个 p^k 就是 g(x, p) ，f(x, n) = g(n, p₁) × g(n, p₂) ×···× g(n，p_n)，prime(x) ={p₁, p₂…, p_n}，p₁, p₂…, p_n表示 x 的质因子。最后求出 f(x, 1) × f(x, 2) × ··· × f(x, n)。
  思路： f(x, 1) × f(x, 2) × ··· × f(x, n) = g(1, p₁) × g(1, p₂) × ··· ×g(1, p_n) × g(2, p₁) × g(2, p₂) × ··· ×g(1, p_n) × ··· × g(n, p₁) × g(n, p₂) × ··· ×g(n, p_n) ，从这个式子可以看出对于p₁的贡献，在[1, n] 中有 p₁，p₁ × 某个数(这些数可以贡献一个p₁)，对于 p₁² 的贡献就是p₁² × 某个数(这些数可以贡献出两个p₁)···综上所述，我们反过来想这么一个过程其实就是在寻找对于[1, n]之间的数字能够贡献多少p₁，p₂···，p_n。对于 n 来说，n / p₁，在这些数之间有 n / p₁个数可以整除 p₁，故有n / p₁ 个数可以贡献出一个 p₁，n / p₁²，在这些数之间有 n / p₁²个数可以整除 p₁，故有n / p₁² 个数可以在之前的基础上贡献出一个 p₁，一次类推。对于p，他出现的次数就是 cnt = n / p + n / p²+ n / p³+ ···· + n / pⁿ，这个值就是 p^cnt。这个问题就这样转换成求快速幂的过程。对于此题找出所有的 p ，计算出次数 快速幂累乘即可解决。

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
const int mod = 1e9 + 7;
const int Max_n = 1e6 + 10;
int prime[Max_n];

ll qpow(ll a, ll b) {
    ll ans = 1, res = a;
    while(b) {
        if(b & 1) ans = ans * res % mod;
        res = res * res % mod;
        b >>= 1;
    }
    return ans % mod;
}

int main() {
    ll x, n, inx = 0;
    scanf("%lld%lld", &x, &n);
    for(int i = 2; i * i <= x; i++) {
        if(!(x % i)) {
            prime[inx++] = i;
            while(!(x % i)) x /= i;
        }
    }
    if(x > 1) prime[inx++] = x;
    ll ans = 1;
    for(int i = 0; i < inx; i++) {
        ll t = n, cnt = 0;
        while(t) {//求次数
            t /= prime[i];
            cnt += t;
        }
        ans = (ans * qpow(prime[i], cnt)) % mod;
    }
    printf("%lld\n", ans);
    return 0;
}