結果

問題 No.1042 愚直大学
コンテスト
ユーザー Theta
提出日時 2022-10-07 15:17:11
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 464 ms / 2,000 ms
コード長 1,184 bytes
コンパイル時間 191 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 44,356 KB
最終ジャッジ日時 2024-06-11 22:13:32
合計ジャッジ時間 14,054 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import deque
from math import log2, log
import numpy as np
from random import random


class DiffSolveTime:
    def __init__(self, p: int, q: int):
        self.__p = p
        self.__q = q

    def diff_solve_time(self, n: int | float) -> float:
        return n ** 2 - self.__p - self.__q * n * log2(n)

    def diff_solve_time_derived(self, n: int | float) -> float:
        return n * 2 - self.__q * log2(n) - self.__q / log(2)

    def diff2_solve_time_derived(self, n: int | float) -> float:
        return 2 - self.__q / (n * log(2))

    def get_inflection(self) -> float:
        return self.__q / (2*log(2))


def main():
    P, Q = map(int, input().split())

    solver = DiffSolveTime(P, Q)
    N = 1e2
    ans_deque = deque(maxlen=2)
    while solver.diff_solve_time(N) < 0:
        N *= 10

    counter = 0
    while abs(solver.diff_solve_time(N)) > 1e-6 and counter < 5:
        if N in ans_deque:
            N = np.mean(ans_deque)
            counter += 1
            continue
        ans_deque.append(N)
        N -= (solver.diff_solve_time(N) /
              solver.diff_solve_time_derived(N))

    print(N)


if __name__ == "__main__":
    main()
0