結果

問題 No.2125 Inverse Sum
コンテスト
ユーザー t98slider
提出日時 2022-11-19 12:20:20
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 18 ms / 2,000 ms
コード長 1,610 bytes
コンパイル時間 2,228 ms
コンパイル使用メモリ 180,424 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-09-20 13:14:47
合計ジャッジ時間 4,048 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

//素因数分解
template <typename T> vector<pair<T, T>> prime_factor(T n) {
    vector<pair<T, T>> ret;
    T tmp=0;
    if(n%2==0){
        tmp=0;while(n%2==0){tmp++;n/=2;}
        ret.push_back(make_pair(2, tmp));
    }
    if(n%3==0){
        tmp=0;while(n%3==0){tmp++;n/=3;}
        ret.push_back(make_pair(3, tmp));
    }
    for (T i=5;i*i<=n;i+=4) {
        if(n%i==0){
            tmp=0;while(n%i==0){tmp++;n/=i;}
            ret.push_back(make_pair(i,tmp));
        }
        i+=2;
        if(n%i==0){
            tmp=0;while(n%i==0){tmp++;n/=i;}
            ret.push_back(make_pair(i,tmp));
        }
    }
    if(n!=1)ret.push_back(make_pair(n,1));
    return ret;
}

int main(){
    ll P, Q, Q2, g;
    cin >> P >> Q;
    g = __gcd(P, Q);
    P /= g, Q /= g;
    auto prime = prime_factor(Q);
    for(auto &&p:prime)p.second <<= 1;
    vector<ll> y;
    function<void(int,ll)> dfs = [&](int i, ll v){
        if(i == prime.size()){
            y.push_back(v);
            return;
        }
        ll d = 1;
        for(int j = 0; j <= prime[i].second; j++){
            if(j)d *= prime[i].first;
            dfs(i + 1, v * d);
        }
    };
    dfs(0, 1);
    vector<pair<ll,ll>> ans;
    Q2 = Q * Q;
    for(auto &&v:y){
        ll v1 = Q2 / v;
        if((v + Q) % P == 0 && (v1 + Q) % P == 0)ans.emplace_back((v + Q) / P, (v1 + Q) / P);
    }
    sort(ans.begin(), ans.end());
    cout << ans.size() << '\n';
    for(int i = 0; i < ans.size(); i++){
        cout << ans[i].first << ' ' << ans[i].second << '\n';
    }
}
0