結果

問題 No.2798 Multiple Chain
ユーザー GOTKAKO
提出日時 2024-06-28 23:06:36
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 2,128 bytes
コンパイル時間 1,924 ms
コンパイル使用メモリ 206,356 KB
最終ジャッジ日時 2025-02-22 01:21:04
ジャッジサーバーID
（参考情報） 		judge2 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 TLE * 1 -- * 1
other -- * 51
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    long long N; cin >> N;
    //N = 897612484786617600LL/37*32/31*32/29*27/23*25;
    //cout << N << endl;
    vector<int> pf;
    {
        int e2 = 0,e3 = 0;
        while(N%2 == 0) e2++,N /= 2;
        while(N%3 == 0) e3++,N /= 3;
        if(e2) pf.push_back(e2);
        if(e3) pf.push_back(e3);
    }
    {
        long long i=5;
        while(i*i <= N){
            int e = 0;
            while(N%i == 0) e++,N /= i;
            if(e) pf.push_back(e);
            e = 0; i += 2;
            while(N%i == 0) e++,N /= i;
            if(e) pf.push_back(e);
            i += 4;
        }
        if(N != 1) pf.push_back(1);
    }


    int answer = 0,n = pf.size();
    vector<int> multi(n+1,1);
    for(int i=1; i<=n; i++){
        int now = pf.at(i-1)+1;
        multi.at(i) = multi.at(i-1)*now;
    }
    swap(pf,multi); //まちがえた.

    unordered_map<long long,int> M;
    auto dfs = [&](auto dfs,int left,int need) -> int {
        int &ret = M[left*1001001001LL+need];
        answer += ret;
        if(ret) return ret;
        for(int i=0; i<n; i++){
            int le = left%pf.at(i+1)/pf.at(i),ne = need%pf.at(i+1)/pf.at(i);
            if(ne > le) return 0;
            left -= ne*pf.at(i);
        }
        answer++; ret++;
        if(left == 0) return ret;

        int next = left;
        vector<int> Left(n),add(n);
        for(int i=0; i<n; i++) Left.at(i) = left%pf.at(i+1)/pf.at(i);

        if(need) ret += dfs(dfs,left,need);
        while(true){
            int pos = -1;
            while(++pos != n){
                if(add.at(pos) == Left.at(pos)){
                    next += Left.at(pos)*pf.at(pos);
                    need -= Left.at(pos)*pf.at(pos);
                    add.at(pos) = 0;
                }
                else{add.at(pos)++; next -= pf.at(pos),need += pf.at(pos); break;}
            }
            if(pos == n) break;
            ret += dfs(dfs,next,need);
        }
        return ret;
    };
    dfs(dfs,pf.at(n)-1,0);
    cout << answer << endl;
}
0