結果

問題 No.3156 Count That Day's N
コンテスト
ユーザー Today03
提出日時 2025-05-23 20:04:37
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 120 ms / 3,000 ms
コード長 3,414 bytes
コンパイル時間 2,906 ms
コンパイル使用メモリ 280,112 KB
実行使用メモリ 7,848 KB
最終ジャッジ日時 2025-05-23 20:04:43
合計ジャッジ時間 4,988 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

#define ALL(x) (x).begin(),(x).end()
#define IO ios::sync_with_stdio(false),cin.tie(nullptr);
#define REP(i, n) for(ll i=0; i<(ll)(n); i++)
#define FOR(i, a, b) for(ll i=(ll)(a); i<(b); i++)
#define ROF(i, a, b) for(ll i=(ll)(b)-1; i>=(b); i--)

template<typename T> int LB(const vector<T>& v, T x) { return lower_bound(ALL(v),x)-(v).begin(); }
template<typename T> int UQ(T& v) { sort(ALL(v)); v.erase(unique(ALL(v)),v.end()); return v.size(); }
template<typename T> bool chmax(T &a, T b) { return a<b ? a=b, true : false; }
template<typename T> bool chmin(T &a, T b) { return a>b ? a=b, true : false; }
template<typename T> using rpriority_queue=priority_queue<T,vector<T>,greater<T>>;
using ll=long long; const int INF=1e9+10; const ll INFL=4e18; using ld=long double;
using ull=unsigned long long; using lll=__int128_t; using VST=vector<string>;
using VI=vector<int>; using VVI=vector<VI>; using VL=vector<ll>; using VVL=vector<VL>;
using PL=pair<ll,ll>; using VP=vector<PL>; using WG=vector<vector<pair<int,ll>>>;

#ifdef LOCAL
#include "./debug.hpp"
#else
#define debug(...)
#define print_line
#endif


/// @brief 二分探索
/// @details 条件 judge を満たす ok と ng の境界を二分探索によって求める。
/// @note O(log(|ok - ng|) * f)
template<typename T, typename Judge>
T BinarySearch(T ok, T ng, Judge judge) {
    while(abs(ok-ng)>1) {
        T mid=(ok+ng)/2;
        if(judge(mid)) ok=mid;
        else ng=mid;
    }
    return ok;
}

/// @brief 回数指定二分探索
/// @details 条件 judge を満たす ok と ng の境界を二分探索によって求める。
/// @note O(iter * f)
template<typename T, typename Judge>
T BinarySearchIteration(T ok, T ng, Judge judge, int iter=100) {
    while(iter--) {
        T mid=(ok+ng)/2;
        if(judge(mid)) ok=mid;
        else ng=mid;
    }
    return ok;
}

/// @brief 単調性の確認
/// @details 関数 judge が単調性を満たすか否かを確認する
/// @param start 開始要素
/// @param step 探索幅
/// @param iter 探索回数
template<typename T, typename Judge>
bool CheckMonotonicity(T start, T step, ll iter, Judge judge) {
    cerr<<"[ ";
    bool pre=false;
    ll cnt=0;
    for(T i=start; iter>0; iter--, i+=step) {
        bool tmp=judge(i);
        cerr<<"{ "<<i<<" : "<<(tmp ? "OK" : "NG")<<" }, ";
        if(i!=start&&tmp!=pre) cnt++;
        pre=tmp;
    }
    cerr<<" ]\n";

    if(cnt<=1) cerr<<"Is Monotonic\n";
    else cerr<<"Not Monotonicss\n";
    return cnt<=1;
}


/// @brief n　の k 乗根の切り捨てを返す
ll KthRoot(ll n, ll k=2) {
    if(k==1) return n;
    return (ll)BinarySearch<ll>(0,4e10,[&](ll x) {
        lll tmp=1;
        REP(i,k) {
            if(tmp*x>n) return false;
            tmp*=x;
        }
        return true;
    });
}

//----------------------------------------------------------

int main() {
    ll K,N; cin>>K>>N;

    set<lll> st;
    for(lll x=1; x*x*x*x*x*x<=N; x++) {
        lll X=x*x*x*x*x*x;
        for(lll y=1; X+y*y*y*y<=N; y++) {
            lll Y=y*y*y*y;

            lll n=(X+Y);
            debug(x,y,n);
            if(n%K==0) {
                lll zz=n/K;
                lll z=KthRoot(zz);
                if(z>0&&z*z==zz) {
                    debug(x,y,z,n,zz);
                    st.insert(n);
                }
            }
        }
    }

    cout<<st.size()<<endl;
}
0