結果
| 問題 | No.3537 Thank You! |
| コンテスト | |
| ユーザー |
|
| 提出日時 | 2026-08-29 14:36:47 |
| 言語 | PyPy3 (7.3.23) |
| 結果 |
AC
|
| 実行時間 | 231 ms / 2,000 ms |
| + 281µs | |
| コード長 | 2,366 bytes |
| 記録 | |
| コンパイル時間 | 565 ms |
| コンパイル使用メモリ | 95,968 KB |
| 実行使用メモリ | 112,896 KB |
| 最終ジャッジ日時 | 2026-08-29 14:36:55 |
| 合計ジャッジ時間 | 7,035 ms |
|
ジャッジサーバーID (参考情報) |
judge1_0 / judge3_0 |
(要ログイン)
| サブタスク | 配点 | 結果 |
|---|---|---|
| サブタスク1 | 30 % | AC * 21 |
| サブタスク2 | 70 % | AC * 15 |
| 合計 | 2.5 * 100% = 250 点 |
ソースコード
# https://yukicoder.me/problems/no/3537
class BinaryIndexTree:
"""
フェニック木(BinaryIndexTree)の基本的な機能を実装したクラス
"""
def __init__(self, size):
self.size = size
self.array = [0] * (size + 1)
def add(self, x, a):
index = x
while index <= self.size:
self.array[index] += a
index += index & (-index)
def sum(self, x):
index = x
ans = 0
while index > 0:
ans += self.array[index]
index -= index & (-index)
return ans
def least_upper_bound(self, value):
if self.sum(self.size) < value:
return -1
elif value <= 0:
return 0
m = 1
while m < self.size:
m *= 2
k = 0
k_sum = 0
while m > 0:
k0 = k + m
if k0 < self.size:
if k_sum + self.array[k0] < value:
k_sum += self.array[k0]
k += m
m //= 2
if k < self.size:
return k + 1
else:
return -1
def main():
N = int(input())
B = int(input())
C = list(map(int, input().split()))
S = list(map(int, input().split()))
c_max = max(C)
bit_count = BinaryIndexTree(c_max)
bit_sum = BinaryIndexTree(c_max)
for i in range(N):
c = C[i]
s = S[i]
bit_count.add(c, s)
bit_sum.add(c, c * s)
# 市場操作をしないケース
# この金額の商品まで全て買おうとすると「初めてBを超える商品の金額」
b_c = bit_sum.least_upper_bound(B)
answer = bit_count.sum(b_c - 1)
x = B - bit_sum.sum(b_c - 1)
y = x // b_c
answer += y
# 市場操作を行うケース
for i in range(N):
c = C[i]
s = S[i]
bit_count.add(c, -s)
bit_sum.add(c, -c * s)
bit_count.add(1, s)
bit_sum.add(1, s)
b_c = bit_sum.least_upper_bound(B)
ans = bit_count.sum(b_c - 1)
x = B - bit_sum.sum(b_c - 1)
y = x // b_c
ans += y
answer = max(answer ,ans)
bit_count.add(1, -s)
bit_sum.add(1, -s)
bit_count.add(c, s)
bit_sum.add(c, c * s)
print(answer)
if __name__ == "__main__":
main()