結果

問題 No.1581 Multiple Sequence
ユーザー KudeKude
提出日時 2021-07-02 22:57:18
言語 C++17
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 45 ms / 2,000 ms
コード長 2,799 bytes
コンパイル時間 4,527 ms
コンパイル使用メモリ 263,152 KB
実行使用メモリ 15,340 KB
最終ジャッジ日時 2023-09-11 23:23:34
合計ジャッジ時間 6,384 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge14
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 7 ms
7,772 KB
testcase_01 AC 7 ms
7,776 KB
testcase_02 AC 38 ms
13,996 KB
testcase_03 AC 40 ms
14,620 KB
testcase_04 AC 17 ms
10,072 KB
testcase_05 AC 41 ms
14,768 KB
testcase_06 AC 24 ms
11,420 KB
testcase_07 AC 18 ms
9,808 KB
testcase_08 AC 20 ms
10,644 KB
testcase_09 AC 38 ms
14,348 KB
testcase_10 AC 29 ms
12,192 KB
testcase_11 AC 13 ms
8,888 KB
testcase_12 AC 22 ms
10,956 KB
testcase_13 AC 8 ms
7,648 KB
testcase_14 AC 41 ms
14,540 KB
testcase_15 AC 31 ms
12,932 KB
testcase_16 AC 14 ms
9,088 KB
testcase_17 AC 44 ms
15,324 KB
testcase_18 AC 12 ms
8,776 KB
testcase_19 AC 36 ms
13,984 KB
testcase_20 AC 38 ms
13,972 KB
testcase_21 AC 40 ms
14,624 KB
testcase_22 AC 27 ms
11,880 KB
testcase_23 AC 45 ms
15,340 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
#include<atcoder/all>
using namespace std;
using namespace atcoder;
#define rep(i,n)for (int i = 0; i < int(n); ++i)
#define rrep(i,n)for (int i = int(n)-1; i >= 0; --i)
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()
template<class T> void chmax(T& a, const T& b) {a = max(a, b);}
template<class T> void chmin(T& a, const T& b) {a = min(a, b);}
using ll = long long;
using P = pair<int,int>;
using VI = vector<int>;
using VVI = vector<VI>;
using VL = vector<ll>;
using VVL = vector<VL>;
using mint = modint1000000007;

std::pair<std::vector<int>,std::vector<int>> primes_lpf(const int n) {
    std::pair<std::vector<int>,std::vector<int>> rv;
    std::vector<int>& primes = rv.first;
    std::vector<int>& lpf = rv.second;
    primes.reserve(n / 10);
    lpf.resize(n + 1);
    for(int i = 2; i <= n; i += 2) lpf[i] = 2;
    for(int i = 3; i <= n; i += 6) lpf[i] = 3;
    if (2 <= n) primes.push_back(2);
    if (3 <= n) primes.push_back(3);
    // 5 * x <= n, x <= floor(n / 5)
    const int n5 = n / 5;
    int x = 5;
    bool one_mod6 = false;
    // x_1 <= n5, x_2 = x_1 + 2 <= n5 + 2 <= n
    for(; x <= n5; one_mod6 = !one_mod6, x += (one_mod6 ? 2 : 4)) {
        if (lpf[x] == 0) {
            lpf[x] = x;
            primes.push_back(x);
        }
        int px = lpf[x];
        for(int i = 2;; ++i) {
            int q = primes[i];
            int y = q * x;
            if (y > n) break;
            lpf[y] = q;
            if (q == px) break;
        }
    }
    for(; x <= n; one_mod6 = !one_mod6, x += (one_mod6 ? 2 : 4)) {
        if (lpf[x] == 0) {
            lpf[x] = x;
            primes.push_back(x);
        }
    }
    return rv;
}

auto [primes, lpf] = primes_lpf(1000000);

std::vector<std::pair<int,int>> factorize(int n) {
    std::vector<std::pair<int,int>> fs;
    while(n > 1) {
        int p = lpf[n];
        int cnt = 0;
        do {n /= p; cnt++;} while(n % p == 0);
        fs.emplace_back(p, cnt);
    }
    return fs;
}

vector<int> divisors(int n) {
    assert(n >= 1);
    vector<int> ret{1};
    while(n > 1) {
        int p = lpf[n];
        int cnt = 0;
        do {n /= p; cnt++;} while(n % p == 0);
        int sz = ret.size();
        for(int q = 1, c = 1; c <= cnt; c++) {
            q *= p;
            for(int i = 0; i < sz; i++) ret.push_back(ret[i] * q);
        }
    }
    return ret;
}

mint memo[100001];
bool valid[100001];
mint f(int rest) {
    if (rest == 0) return 1;
    if (valid[rest]) return memo[rest];
    mint ret;
    for(int d: divisors(rest)) {
        ret += f(d - 1);
    }
    valid[rest] = true;
    return memo[rest] = ret;
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    int m;
    cin >> m;
    mint ans = f(m);
    cout << ans.val() << endl;
}
0