結果

問題 No.1666 累乗数
ユーザー ruthenruthen
提出日時 2021-09-04 01:48:46
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 115 ms / 2,000 ms
コード長 4,561 bytes
コンパイル時間 3,509 ms
コンパイル使用メモリ 170,784 KB
実行使用メモリ 12,588 KB
最終ジャッジ日時 2023-08-22 04:50:47
合計ジャッジ時間 5,029 ms
ジャッジサーバーID
(参考情報)
judge15 / judge12
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 107 ms
12,064 KB
testcase_01 AC 112 ms
11,728 KB
testcase_02 AC 112 ms
11,760 KB
testcase_03 AC 112 ms
12,352 KB
testcase_04 AC 111 ms
11,448 KB
testcase_05 AC 112 ms
11,808 KB
testcase_06 AC 113 ms
12,416 KB
testcase_07 AC 112 ms
12,548 KB
testcase_08 AC 112 ms
12,588 KB
testcase_09 AC 115 ms
11,616 KB
testcase_10 AC 113 ms
11,424 KB
testcase_11 AC 113 ms
11,620 KB
testcase_12 AC 113 ms
11,628 KB
testcase_13 AC 114 ms
11,616 KB
testcase_14 AC 113 ms
11,616 KB
testcase_15 AC 114 ms
11,556 KB
testcase_16 AC 113 ms
11,168 KB
testcase_17 AC 113 ms
12,316 KB
testcase_18 AC 114 ms
11,920 KB
testcase_19 AC 112 ms
11,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

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

// clang-format off
//#include <atcoder/all>
//using namespace atcoder;
using ll = long long;
using ld = long double;
using pii = pair<int, int>;
using pll = pair<ll, ll>;
using pdd = pair<ld, ld>;
using vii = vector<int>;
using vll = vector<ll>;
using vdd = vector<ld>;
using vvii = vector<vector<int>>;
using vvll = vector<vector<ll>>;
using vvdd = vector<vector<ld>>;
using vvvii = vector<vector<vector<int>>>;
using vvvll = vector<vector<vector<ll>>>;
using vvvdd = vector<vector<vector<ld>>>;
template<class T, class U> using P = pair<T, U>;
template<class T> using V1 = vector<T>;
template<class T> using V2 = vector<vector<T>>;
template<class T> using V3 = vector<vector<vector<T>>>;
template<class T> using pque = priority_queue<T, vector<T>, greater<T>>;
#define _overload3(_1, _2, _3, name, ...) name
#define rep1(n) for (ll i = 0; i < (n); i++)
#define rep2(i, n) for (ll i = 0; i < (n); i++)
#define rep3(i, a, b) for (ll i = (a); i < (b); i++)
#define rep(...) _overload3(__VA_ARGS__, rep3, rep2, rep1)(__VA_ARGS__)
#define rrep1(n) for (ll i = (n) - 1; i >= 0; i--)
#define rrep2(i, n) for (ll i = (n) - 1; i >= 0; i--)
#define rrep3(i, a, b) for (ll i = (b) - 1; i >= (a); i--)
#define rrep(...) _overload3(__VA_ARGS__, rrep3, rrep2, rrep1)(__VA_ARGS__)
#define all(x) (x).begin(), (x).end()
#define sz(x) ((int)(x).size())
#define fi first
#define se second
#define pb push_back
#define endl '\n'
#define popcnt(x) __builtin_popcountll(x)
#define uniq(x) (x).erase(unique((x).begin(), (x).end()), (x).end());
#define IOS ios::sync_with_stdio(false); cin.tie(nullptr);
const int inf = 1 << 30;
const ll INF = 1ll << 60;
const ld DINF = numeric_limits<ld>::infinity();
const ll mod = 1000000007;
//const ll mod = 998244353;
const ld EPS = 1e-9;
const ld PI = 3.1415926535897932;
const ll dx[8] = {1, 0, -1, 0, 1, -1, -1, 1};
const ll dy[8] = {0, 1, 0, -1, 1, 1, -1, -1};
template<class T> bool chmax(T &a, const T &b) { if (a<b) { a=b; return 1; } return 0; }
template<class T> bool chmin(T &a, const T &b) { if (a>b) { a=b; return 1; } return 0; }
template<typename T> T min(const vector<T> &x) { return *min_element(all(x)); }
template<typename T> T max(const vector<T> &x) { return *max_element(all(x)); }
template<class T, class U> ostream &operator<<(ostream &os, const pair<T, U> &p) { return os << "(" << p.fi << ", " << p.se << ")"; }
template<class T> ostream &operator<<(ostream &os, const vector<T> &v) { os << "[ "; for (auto &z : v) os << z << " "; os << "]"; return os; }
#ifdef _DEBUG
#define show(x) cout << #x << " = " << x << endl;
#else
#define show(x)
#endif
ll gcd(ll a, ll b) { return b ? gcd(b, a % b) : a; }
ll lcm(ll a, ll b) { return (a / gcd(a, b)) * b; }
ll rem(ll a, ll b) { return (a % b + b) % b; }
// clang-format on

// using mint = modint998244353;

ll modpow(ll a, ll N, ll mod) {
    ll res = 1;
    // 例えば3=101(2)なので、下位bitから順に1ならa倍する
    while (N) {
        if (N & 1) res = res * a % mod;
        a = a * a % mod;
        N >>= 1;
    }
    return res;
}

bool isRoot(ll x) {
    ld sq = sqrt(x);
    ll res = (ll)sq;
    return (res * res == x);
}

ll mysqrt(ll x) {
    ll lb = 1, ub = inf;
    while (ub - lb > 1) {
        ll mid = (lb + ub) / 2;
        if (mid * mid <= x)
            lb = mid;
        else
            ub = mid;
    }
    return lb;
}

void solve(vll &b3, ll k) {
    // x以下の累乗数がcnt(x)個ある
    // cnt(x) >= Kとなるxの最小値
    // (lb, ub]
    ll ub = INF, lb = 0;
    while (ub - lb > 1) {
        ll mid = (ub + lb) / 2;
        ll cnt = upper_bound(all(b3), mid) - b3.begin();
        cnt += mysqrt(mid);
        if (cnt >= k)
            ub = mid;
        else
            lb = mid;
    }
    cout << ub << endl;
    return;
}

int main() {
    vll b3;
    rep(a, 2, 1000005) {
        ll x = a * a * a;
        while (1) {
            if (!isRoot(x)) b3.pb(x);
            if (x > INF / a) break;
            x *= a;
        }
    }
    sort(all(b3));
    uniq(b3);
    int t;
    cin >> t;
    while (t--) {
        ll k;
        cin >> k;
        solve(b3, k);
    }
    return 0;
}

/*
メモ
二分探索
b>=3の場合でa^2の形で表せないものを前もって列挙する
b>=3よりa<=10^6まで考えれば良いので全部列挙できる
ある数xより小さい累乗数は
b=2のものはsqrt(x)で、b>=3は全列挙したものから二分探索することで求められる
sqrt(x)がxが大きいとめちゃくちゃ誤差が出る
自作する
*/
0