結果
問題 | No.1868 Teleporting Cyanmond |
ユーザー |
|
提出日時 | 2022-03-11 22:04:36 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 362 ms / 2,000 ms |
コード長 | 2,187 bytes |
コンパイル時間 | 203 ms |
コンパイル使用メモリ | 82,320 KB |
実行使用メモリ | 96,384 KB |
最終ジャッジ日時 | 2024-09-16 02:21:53 |
合計ジャッジ時間 | 8,792 ms |
ジャッジサーバーID (参考情報) |
judge6 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 25 |
ソースコード
class SegTree:"""init(init_val, ide_ele): 配列init_valで初期化 O(N)update(k, x): k番目の値をxに更新 O(logN)query(l, r): 区間[l, r)をsegfuncしたものを返す O(logN)"""def __init__(self, init_val, segfunc, ide_ele):"""init_val: 配列の初期値segfunc: 区間にしたい操作ide_ele: 単位元n: 要素数num: n以上の最小の2のべき乗tree: セグメント木(1-index)"""n = len(init_val)self.segfunc = segfuncself.ide_ele = ide_eleself.num = 1 << (n - 1).bit_length()self.tree = [ide_ele] * 2 * self.num# 配列の値を葉にセットfor i in range(n):self.tree[self.num + i] = init_val[i]# 構築していくfor i in range(self.num - 1, 0, -1):self.tree[i] = self.segfunc(self.tree[2 * i], self.tree[2 * i + 1])def update(self, k, x):"""k番目の値をxに更新k: index(0-index)x: update value"""k += self.numself.tree[k] = xwhile k > 1:self.tree[k >> 1] = self.segfunc(self.tree[k], self.tree[k ^ 1])k >>= 1def query(self, l, r):"""[l, r)のsegfuncしたものを得るl: index(0-index)r: index(0-index)"""res = self.ide_elel += self.numr += self.numwhile l < r:if l & 1:res = self.segfunc(res, self.tree[l])l += 1if r & 1:res = self.segfunc(res, self.tree[r - 1])l >>= 1r >>= 1return resdef value(self,index):return self.query(index,index+1)N = int(input())R = [0] + list(map(int,input().split()))dp = [0 for _ in range(N+1)]dp[1] = 0INF = float('inf')tmp = [INF for _ in range(N+1)]seg = SegTree(tmp,segfunc=lambda x,y:min(x,y),ide_ele=INF)seg.update(R[1],dp[1])for i in range(2,N+1):dp[i] = seg.query(i,N+1) + 1if i == N:continueseg.update(R[i],min(seg.value(R[i]),dp[i]))print(dp[N])