#674728 (Python3) No.1590 Random Shopping

• 問題: No.1590 Random Shopping
• ユーザー: 👑 SPD_9X2
• 提出日時: 2021-07-09 18:20:14
• 言語: Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1)
• 結果: TLE
• 実行時間: -
• コード長: 1,843 bytes
• コンパイル時間: 154 ms
• コンパイル使用メモリ: 12,672 KB
• 実行使用メモリ: 275,200 KB
• 最終ジャッジ日時: 2024-07-01 14:02:24
• 合計ジャッジ時間: 9,586 ms
• ジャッジサーバーID （参考情報）: judge1 / judge3

ファイルパターン	結果
sample	-- * 3
other	AC * 12 TLE * 1 -- * 12

ソースコード

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
"""
https://yukicoder.me/problems/no/1590
分からなければさっさと見ちゃおうかな
寄与を考えるよね
x人目がy番目の商品を買う確率
がわかればおk
x,yのペアに関して求めたい
=====================
答えを見た
出来そうなdpから結果を導出する
Borderごとに
dp[Border][a][b] = a日目でBorder以下がb個残っている確率
とすると、
a日目の人の購入額の期待値は
1/2 * ( dp[B0][a][0] * B0 + dp[B1][a][0] * (B1-B0) + ... )
ただし、その日の商品 A[a] < B?
p[i][j] = 
"""
from sys import stdin
N = int(stdin.readline())
A = list(map(int,stdin.readline().split()))
R = list(map(int,stdin.readline().split()))
s = set(A)
s.add(0)
B = list(s)
B.sort()
#print (B)
dp = {}
for Border in B:
    ndp = [[0] * (N+1) for i in range(N+1)]
    ndp[0][0] = 1
    for a in range(N):
        na = A[a]
        for b in range(N+1):
            if na <= Border:
                
                #追加して、取る場合
                ndp[a+1][b] += ndp[a][b] / 2
                #追加して、取らない場合
                if b != N:
                    ndp[a+1][b+1] += ndp[a][b] / 2
            else:
                #追加はしない、取る場合
                ndp[a+1][max(0,b-1)] += ndp[a][b]/2
                #追加はしない、取らない場合
                ndp[a+1][b] += ndp[a][b] / 2
    dp[Border] = ndp
#print (dp)
ans = 0
for a in range(1,N+1):
    na = A[a-1] 
    nr = R[a-1] 
    nsum = 0
    for j in range(len(B)-1):
        p = 0 #人aが、B[j]より大きい奴を取る確率
        if na <= B[j]:
            p = 0
        else:
            p = dp[B[j]][a-1][0] / 2
        #print (a,B[j],p)
        nsum += (B[j+1]-B[j]) * p
    ans += nsum * nr
    #print (nsum * nr)
print (ans)
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX