結果

問題 No.2368 I love a square root of 2
ユーザー hitonanodehitonanode
提出日時 2023-06-30 22:46:32
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 35 ms / 2,000 ms
コード長 1,174 bytes
コンパイル時間 1,193 ms
コンパイル使用メモリ 119,280 KB
実行使用メモリ 5,584 KB
最終ジャッジ日時 2024-07-07 10:38:12
合計ジャッジ時間 2,075 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cmath>
#include <algorithm>
#include <iostream>
#include <tuple>
#include <vector>
using namespace std;
using lint = long long;
#define FOR(i, begin, end) for(int i=(begin),i##_end_=(end);i<i##_end_;i++)
#define REP(i, n) FOR(i,0,n)

int main() {
    lint N;
    cin >> N;

    int ng = -1, ok = 1e6;

    lint fng = 0;

    const double sqrt2 = sqrt(2.0);

    auto f = [&](int l) {
        lint ret = 0;
        REP(x, l + 1) {
            double u = l - sqrt2 * x;
            if (u >= 0) ret += u + 1;
            else {
                break;
            }
        }
        return ret;
    };

    while (ok - ng > 1) {
        const lint c = (ok + ng) / 2;
        if (auto fc = f(c); fc >= N) {
            ok = c;
        } else {
            fng = fc, ng = c;
        }
    }

    const lint req = N - fng;
    vector<tuple<double, int, int>> vs;
    REP(x, ok + 1) {
        int y = ok - sqrt2 * x;
        if (y < 0) break;
        if (auto t = x * sqrt2 + y; t > ok - 1) vs.emplace_back(t, x, y);
    }

    partial_sort(vs.begin(), vs.begin() + req, vs.end());

    auto ret = vs.at(req - 1);
    cout << get<2>(ret) << ' ' << get<1>(ret) << '\n';
}
0