結果

問題 No.1666 累乗数
ユーザー karinohito
提出日時 2021-09-03 23:42:36
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 297 ms / 2,000 ms
コード長 1,741 bytes
コンパイル時間 2,005 ms
コンパイル使用メモリ 178,892 KB
実行使用メモリ 60,080 KB
最終ジャッジ日時 2024-12-15 18:49:48
合計ジャッジ時間 8,716 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
//#include<atcoder/all>
using namespace std;
//using namespace atcoder;
using ll = long long;
#define all(A) A.begin(),A.end()
using vll = vector<ll>;
#define rep(i, n) for (long long i = 0; i < (long long)(n); i++)
using Graph = vector<vector<ll>>;
vll P = { 3,5,7,11,13,17,19,23,29,31,37 };
vector<vll> PP(P.size());
set<ll> S;
vll PPP;
ll sqrtz(ll N) {
    ll L = 0;
    ll R = sqrt(N)+10000;
    while (abs(R - L) > 1) {
        ll mid = (R +L) / 2;
        if (mid * mid <= N)L = mid;
        else R = mid;
    }
    return L;
    
}



bool isOK(ll index, ll key) {

    ll k = 0;
    ll D = sqrtz(index);
    k += D;



    auto Ds = upper_bound(all(PPP), index) - PPP.begin();

    k += (Ds);

    return (k >= key);


}

// 汎用的な二分探索のテンプレ
ll binary_search(ll key) {
    ll ng = 0; //「index = 0」が条件を満たすこともあるので、初期値は -1
    ll ok = key * key + 3; // 「index = a.size()-1」が条件を満たさないこともあるので、初期値は a.size()

    /* ok と ng のどちらが大きいかわからないことを考慮 */
    while (abs(ok - ng) > 1) {
        ll mid = (ok + ng) / 2;
        cout << "";
        if (isOK(mid, key)) ok = mid;
        else ng = mid;
    }
    return ok;
}

int main() {
    //set<ll> S;
    S.insert(2e18);
    for (ll p = 2; p <= 1e6+4; p++) {
        ll k = sqrtz(p);
        if (k * k == p)continue;
        ll d = p;
        while (d <= (1e18 + 6) / (p * p)) {
            d *= p * p;
            S.insert(d);
        }
    }
    for (auto l : S) {
        PPP.push_back(l);
    }
    ll T;
    cin >> T;
    rep(t, T) {
        ll N;
        cin >> N;
        cout << binary_search(N) << endl;
    }
}
0