結果

問題 No.5018 Let's Make a Best-seller Book
ユーザー すぎぱも
提出日時 2023-10-01 14:47:28
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 827 bytes
コンパイル時間 276 ms
コンパイル使用メモリ 87,236 KB
実行使用メモリ 88,092 KB
スコア 0
平均クエリ数 2.00
最終ジャッジ日時 2023-10-01 14:47:44
合計ジャッジ時間 15,290 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge12
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other WA * 100
権限があれば一括ダウンロードができます

ソースコード

diff # 

T, N, money = map(int, input().split())
s, p, r = [0] * N, [0] * N, [0] * N


def softmax(x):
    import math
    max_x = max(x)
    exp_x = [e - max_x for e in x]
    exp_x = [math.exp(e) for e in exp_x]
    softmax_values = [e / sum(exp_x) for e in exp_x]
    return softmax_values


def do(arg):
    if type(arg) == list:
        print(1, *arg, flush=True)
    else:
        print(2, arg, flush=True)
    money = int(input())
    if money == -1:
        exit()
    s = list(map(int, input().split()))
    p = list(map(int, input().split()))
    r = list(map(int, input().split()))
    return money, s, p, r


money, s, p, r = do([money // (500 * N)] * N)

for week in range(T - 1):
    pay = [money // 3 * s for s in softmax(p)]
    pay = [int(p) for p in pay]
    money, s, p, r = do(pay)


# 20 20 20 20 20 20 20 20 20 20
0