結果
問題 |
No.1676 Coin Trade (Single)
|
ユーザー |
![]() |
提出日時 | 2021-09-10 22:10:18 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,545 bytes |
コンパイル時間 | 223 ms |
コンパイル使用メモリ | 81,920 KB |
実行使用メモリ | 81,596 KB |
最終ジャッジ日時 | 2024-06-12 00:28:10 |
合計ジャッジ時間 | 5,903 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 WA * 1 |
other | AC * 12 WA * 23 |
ソースコード
import sys input = sys.stdin.readline N, K = map(int, input().split()) dp = [0] * (N + 1) a = [0] * (N + 1) class SegTree: def segfunc(self, x, y): return max(x, y) def __init__(self, n, ide_ele, init_val): #####単位元###### self.ide_ele = ide_ele #num:n以上の最小の2のべき乗 self.num = 2 ** (n - 1).bit_length() self.seg = [self.ide_ele] * 2 * self.num #set_val for i in range(n): self.seg[i + self.num - 1] = init_val[i] #built for i in range(self.num - 2, -1, -1) : self.seg[i] = self.segfunc(self.seg[2 * i + 1], self.seg[2 * i + 2]) def update(self, k, x): k += self.num - 1 self.seg[k] = x while k + 1: k = (k - 1) // 2 self.seg[k] = self.segfunc(self.seg[k * 2 + 1], self.seg[k * 2 + 2]) def query(self, p, q): if q <= p: return self.ide_ele p += self.num - 1 q += self.num - 2 res = self.ide_ele while q - p > 1: if p & 1 == 0: res = self.segfunc(res, self.seg[p]) if q & 1 == 1: res = self.segfunc(res, self.seg[q]) q -= 1 p = p // 2 q = (q - 1) // 2 if p == q: res = self.segfunc(res, self.seg[p]) else: res = self.segfunc(self.segfunc(res, self.seg[p]), self.seg[q]) return res seg = SegTree(N + 1, 0, [0] * (N + 1)) for i in range(1, N + 1): x, m = map(int, input().split()) a[i] = x b = list(map(int, input().split())) for j in b: seg.update(i, max(seg.query(i, i + 1), seg.query(0, j) + x - a[j])) print(seg.query(0, N))