結果

問題 No.691 E869120 and Constructing Array 5
コンテスト
ユーザー tsutaj
提出日時 2018-05-19 21:25:01
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 234 ms / 1,000 ms
コード長 4,103 bytes
コンパイル時間 1,425 ms
コンパイル使用メモリ 125,464 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-06-28 14:50:31
合計ジャッジ時間 8,892 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

// 基本テンプレート
 
#include <iostream>
#include <iomanip>
#include <cstdio>
#include <string>
#include <cstring>
#include <deque>
#include <list>
#include <queue>
#include <stack>
#include <vector>
#include <utility>
#include <algorithm>
#include <map>
#include <set>
#include <complex>
#include <cmath>
#include <limits>
#include <cfloat>
#include <climits>
#include <ctime>
#include <cassert>
#include <numeric>
#include <fstream>
#include <functional>
#include <random>
using namespace std;
 
#define rep(i,a,n) for(int (i)=(a); (i)<(n); (i)++)
#define repq(i,a,n) for(int (i)=(a); (i)<=(n); (i)++)
#define repr(i,a,n) for(int (i)=(a); (i)>=(n); (i)--)
#define debug(...) fprintf(stderr, __VA_ARGS__)
#define int long long int
 
template<typename T> void chmax(T &a, T b) {a = max(a, b);}
template<typename T> void chmin(T &a, T b) {a = min(a, b);}
template<typename T> void chadd(T &a, T b) {a = a + b;}
 
typedef pair<int, int> pii;
typedef long long ll;
 
int dx[] = {0, 0, 1, -1};
int dy[] = {1, -1, 0, 0};
const ll INF = 1001001001001001LL;
const ll MOD = 1000000007LL;

struct Elem {
    long double sum;
    vector<int> nums;
    bool operator<(const Elem &e) const {
        return sum < e.sum;
    }
};

// [lb, ub] の閉区間内の値をランダムに返す構造体
// #include <random> しよう

struct Rand {
public:
    Rand() = default;
    Rand(std::mt19937::result_type seed) : eng(seed) {}
    int NextInt(int lb, int ub) {
        return std::uniform_int_distribution<int>{lb, ub}(eng);
    }
    ll NextLong(ll lb, ll ub) {
        return std::uniform_int_distribution<ll>{lb, ub}(eng);
    }
    double NextDouble(double lb, double ub) {
        return std::uniform_real_distribution<double>{lb, ub}(eng);
    }
private:
    std::mt19937 eng{std::random_device{}()};
};
Rand rnd(114514);

pair< vector<int>, bool > solve(long double target) {
    vector<Elem> cand;
    long double T = target / 8;
    long double EPS = 1e-4;

    // target / 8 に近いもの
    for(int i=0; i<100; ) {
        int P = (int)T;
        int B = rnd.NextInt(0, P*P);

        long double rest = T - sqrtl(B);
        long double flt_A = rest * rest;

        for(int k=0; k<2; k++) {
            int A = (int)flt_A + k;
            long double sum = sqrtl(A) + sqrtl(B);
            if(abs(sum - T) < EPS) {
                cand.push_back(Elem{sum, vector<int>{A, B}});
                i++;
            }
        }
    }

    for(int i=0; i<3; i++) {
        T *= 2, EPS *= 1e-2;
        sort(cand.begin(), cand.end());
        vector<Elem> nxt_cand;

        for(size_t k=0; k<cand.size(); k++) {
            long double diff = T - cand[k].sum;
            long double lb = diff - EPS, ub = diff + EPS;
            
            int it = lower_bound(cand.begin(), cand.end(), Elem{lb, vector<int>()}) - cand.begin();

            // printf("lb = %Lf, ub = %Lf, it = %lld\n", lb, ub, it);
            for(int j=-1; j<2; j++) {
                int x = it - j; if(x < 0 || x >= cand.size()) continue;
                long double val = cand[k].sum + cand[x].sum;
                // printf("val = %Lf, T = %Lf\n", val, T);
                if(abs(val - T) < EPS) {
                    vector<int> nxt_vec;
                    copy(cand[k].nums.begin(), cand[k].nums.end(), back_inserter(nxt_vec));
                    copy(cand[x].nums.begin(), cand[x].nums.end(), back_inserter(nxt_vec));
                    nxt_cand.push_back(Elem{val, nxt_vec});
                }
            }
        }

        swap(cand, nxt_cand);
    }

    if(cand.size() > 0) {
        return make_pair(cand[0].nums, true);
    }
    else {
        return make_pair(vector<int>(), false);
    }
}
 
signed main() {
    int Q; scanf("%lld", &Q);
    for(int t=0; t<Q; t++) {
        long double P; scanf("%Lf", &P);
        vector<int> ans;
        bool found = false;
        while(1) if(tie(ans, found) = solve(P), found) break;
        // solve(P);
        printf("%zu\n", ans.size());
        for(size_t i=0; i<ans.size(); i++) printf("%lld%c", ans[i], i+1==ans.size() ? '\n' : ' ');
    }
    return 0;
}
0