結果

問題 No.691 E869120 and Constructing Array 5
コンテスト
ユーザー hos.lyric
提出日時 2020-09-27 15:39:35
言語 D
(dmd 2.109.1)
結果
AC  
実行時間 768 ms / 1,000 ms
コード長 3,470 bytes
コンパイル時間 2,358 ms
コンパイル使用メモリ 172,776 KB
実行使用メモリ 77,148 KB
最終ジャッジ日時 2024-06-22 09:11:30
合計ジャッジ時間 26,245 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std.conv, std.functional, std.range, std.stdio, std.string;
import std.algorithm, std.array, std.bigint, std.bitmanip, std.complex, std.container, std.math, std.mathspecial, std.numeric, std.regex, std.typecons;
import core.bitop;

class EOFException : Throwable { this() { super("EOF"); } }
string[] tokens;
string readToken() { for (; tokens.empty; ) { if (stdin.eof) { throw new EOFException; } tokens = readln.split; } auto token = tokens.front; tokens.popFront; return token; }
int readInt() { return readToken.to!int; }
long readLong() { return readToken.to!long; }
real readReal() { return readToken.to!real; }

bool chmin(T)(ref T t, in T f) { if (t > f) { t = f; return true; } else { return false; } }
bool chmax(T)(ref T t, in T f) { if (t < f) { t = f; return true; } else { return false; } }

int binarySearch(alias pred, T)(in T[] as) { int lo = -1, hi = cast(int)(as.length); for (; lo + 1 < hi; ) { const mid = (lo + hi) >> 1; (unaryFun!pred(as[mid]) ? hi : lo) = mid; } return hi; }
int lowerBound(T)(in T[] as, T val) { return as.binarySearch!(a => (a >= val)); }
int upperBound(T)(in T[] as, T val) { return as.binarySearch!(a => (a > val)); }

uint xrand() {
  static uint x = 314159265, y = 358979323, z = 846264338, w = 327950288;
  uint t = x ^ x << 11; x = y; y = z; z = w; return w = w ^ w >> 19 ^ t ^ t >> 8;
}

enum N = 30;
enum EPS = 0.9e-10L;
enum MIN_P = 8000.0L;

enum LIM_E = cast(int)(floor((MIN_P / (N - 1))^^2));
enum M = 4 * 10^^5;

void main() {
  alias Entry = Tuple!(double, "r", int[], "es");
  auto ps = new Entry[M];
  auto qs = new Entry[M];
  foreach (i; 0 .. M) {
    ps[i].r = 0.0L;
    ps[i].es = new int[N / 2 - 1];
    foreach (k; 0 .. N / 2 - 1) {
      const e = cast(int)(xrand() % (LIM_E + 1));
      ps[i].r += sqrt(cast(double)(e));
      ps[i].es[k] = e;
    }
    ps[i].r %= 1.0L;
  }
  foreach (i; 0 .. M) {
    qs[i].r = 0.0L;
    qs[i].es = new int[N - N / 2];
    foreach (k; 0 .. N - N / 2) {
      const e = cast(int)(xrand() % (LIM_E + 1));
      qs[i].r += sqrt(cast(double)(e));
      qs[i].es[k] = e;
    }
    qs[i].r %= 1.0L;
  }
  ps.sort!"a.r < b.r";
  qs.sort!"a.r < b.r";
  
  try {
    for (; ; ) {
      const numCases = readInt();
      foreach (caseId; 0 .. numCases) {
        const P = cast(double)(readReal());
        foreach (s; 0 .. 2) {
          const target = s + P % 1.0L;
          for (int i = 0, j = qs.upperBound(Entry(target + EPS - ps[0].r, null)) - 1; i < M && j >= 0; ) {
            if (ps[i].r + qs[j].r < target - EPS) {
              ++i;
            } else if (ps[i].r + qs[j].r > target + EPS) {
              --j;
            } else {
              auto ans = [0] ~ ps[i].es ~ qs[j].es;
              double p = P;
              foreach (k; 1 .. N) {
                p -= sqrt(cast(double)(ans[k]));
              }
              ans[0] = (cast(int)(p + 0.5L))^^2;
              write(N);
              foreach (k; 0 .. N) {
                write(" ", ans[k]);
              }
              writeln();
              debug {
                real sum = 0.0L;
                foreach (k; 0 .. N) {
                  sum += sqrt(cast(real)(ans[k]));
                }
                writefln("%.15f %.15f", sum, sum - P);
                assert(abs(sum - P) <= 1e-10L);
              }
              goto found;
            }
          }
        }
        assert(false, format("FAIL: P = %.15f", P));
       found:
      }
    }
  } catch (EOFException e) {
  }
}
0