結果

問題 No.2798 Multiple Chain
ユーザー GOTKAKO
提出日時 2024-06-28 22:44:19
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,960 bytes
コンパイル時間 2,153 ms
コンパイル使用メモリ 204,480 KB
最終ジャッジ日時 2025-02-22 01:18:27
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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;
    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,long long> M;
    auto dfs = [&](auto dfs,int left,int need) -> long long {
        long long &ret = M[1001001001LL*left+need];
        if(ret){answer += 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++;

        int next = left;
        if(need) ret += dfs(dfs,left,need);

        while(true){
            int pos = 0;
            while(pos != n){
                int now = next%pf.at(pos+1)/pf.at(pos);
                if(now == 0){
                    next += left%pf.at(pos+1)/pf.at(pos)*pf.at(pos);
                    need -= left%pf.at(pos+1)/pf.at(pos)*pf.at(pos);
                    pos++;
                }
                else{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