数论预备知识

常见积性函数

φ(n) －欧拉函数，计算与n互质的正整数之数目　

μ(n) －莫比乌斯函数，关于非平方数的质因子数目

gcd(n,k) －最大公因子，当k固定的情况　

d(n) －n的正因子数目　

σ(n) －n的所有正因子之和　

σk(n) － 因子函数，n的所有正因子的k次幂之和，当中k可为任何复数。　

Idk(n) －幂函数，对于任何复数、实数k，定义为Idk(n) = n^k （完全积性）　

λ(n) －刘维尔函数，关于能整除n的质因子的数目　

γ(n)，定义为γ(n)=(-1)^ω(n)，在此加性函数ω(n)是不能整除n的质数的数目

int gcd(int x, int y) {
    return y == 0 ? x : gcd(y, x % y);
}

// 求解ax + by = 1, 返回gcd(a, b)
// 解的大小|x| <= b, |y| <= a
int extgcd(int a, int b, int &x, int &y) {
    int d = a;
    if (b) {
        d = extgcd(b, a % b, y , x);
        y -= (a / b) * x;
    } else {
        x = 1;
        y = 0;
    }
    return d;
}

扩展欧几里德算法证明

//求解逆元
int mod_inverse(int a, int m) {
    int x, y;
    extgcd(a, m, x, y);
    return (m + x % m) % m;
}

//求欧拉函数值 O(√n)
int euler_phi(int n) {
    int res = n;
    for (int i = 2; i * i <= n; i++) {
        if (n % i == 0) {
            res -= res / i;
            while (n % i == 0) n /= i;
        }
    }
    if (n != 1) res -= res / n;
    return res;
}

// O(maxn)时间内筛出欧拉函数值的表
int euler[maxn];
void euler_pji2() {
    for (int i = 0; i < maxn; i++) euler[i] = i;
    for (int i = 2; i < maxn; i++) {
        if (euler[i] == i) {
            for (int j = i; j < maxn; j += i) {
                euler[j] -= euler[j] / i;
            }
        }
    }
}

// 线性同余方程组
// a¡x ≡ b¡(mod m¡)
// 返回一个(b, m)的数对
pair<int, int> linear_congruence(const vector<int>& A, const vector<int>& B, const vector<int>& M) {
    //由于最开始没有任何限制，所以先把解设为表示所有整数的x£0(mod 1)
    int x = 0, m = 1;
    
    for (int i = 0; i < A.size(); i++) {
        int a = A[i] * m, b = B[i] - A[i] * x, d = gcd(M[i], a);
        if (b % d != 0) return make_pair(0, -1); // 无解
        int t = b / d * mod_inverse(a / d, (M[i] / d) % (M[i] / d));
        x += m * t;
        m *= M[i] / d;
    }
    return make_pair(x % m, m);
}

// 中国剩余定理
// f(x)≡0(mod n) 等价于 f(x)≡0(mod pª) (pª|n)

//n! O(log n)
int fact[maxp];
//分解n!≡b pª, 返回b mod p
int mod_fact(int n, int p, int &a) {
    a = 0;
    if (n == 0) return 1;
    int res = mod_fact(n / p, p, a);
    a += n / p;
    if (n / p % 2 != 0) return res * (p - fact[n % p]) % p;
    return res * fact[n % p] % p;
}

//求组合数
int mod_comb(int n, int k, int p) {
    if (n < 0 || k < 0 || n < k) return 0;
    int e1, e2, e3;
    int a1 = mod_fact(n, p , e1), a2 = mod_fact(k, p, e2), a3 = mod_fact(n - k, p, e3);
    if (e1 > e2 + e3) return 0;
    return a1 * mod_inverse(a2 * a3 % p, p) % p;
}