python/TangDou/Topic/SpareTire.cpp

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define endl "\n"

const int mod = 1e9 + 7; // 尽量这样定义mod ，减少非必要的麻烦

// 快速幂
int qmi(int a, int b) {
    int res = 1;
    a %= mod;
    while (b) {
        if (b & 1) res = res * a % mod;
        b >>= 1;
        a = a * a % mod;
    }
    return res;
}

vector<int> p; // 将m拆分成的质数因子序列p

signed main() {
#ifndef ONLINE_JUDGE
    freopen("SpareTire.in", "r", stdin);
#endif
    int n, m;

    int rev6 = qmi(6, mod - 2);
    int rev2 = qmi(2, mod - 2);

    while (cin >> n >> m) {
        int res = n * (n + 1) % mod * (2 * n + 1) % mod * rev6 % mod;
        int t = n * (n + 1) % mod * rev2 % mod;

        res = (res + t) % mod;

        // 质因子分解
        t = m; // 复制出来
        for (int i = 2; i * i <= t; i++) {
            if (t % i == 0) {
                p.push_back(i);
                while (t % i == 0) t = t / i;
            }
        }
        if (t > 1) p.push_back(t);

        /*
          容斥原理
          例如有3个因子，那么1<<3=8(1000二进制)
          然后i从1开始枚举直到7(111二进制），i中二进制的位置1表式取这个位置的因子
          例如i=3(11二进制) 表示取前两个因子，i=5（101）表示取第1个和第3个的因子
        */
        for (int i = 1; i < (1 << p.size()); i++) {
            int cnt = 0, tmp = 1;
            for (int j = 0; j < p.size(); j++)
                if ((i >> j) & 1) {
                    cnt++;
                    tmp *= p[j];
                }

            int nn = n / tmp;
            int pt = (nn) % mod * (nn + 1) % mod * (2 * nn + 1) % mod * rev6 % mod;
            pt = pt * tmp % mod * tmp % mod;
            pt = (pt + nn * (tmp + tmp * nn) % mod * rev2 % mod) % mod;
            if (cnt & 1)
                res = (res - pt + mod) % mod;
            else
                res = (res + pt) % mod;
        }
        // 补集，标准取模动作
        cout << res << endl;
    }
}