結果
問題 | No.1892 Extended Fib Series |
ユーザー | achapi |
提出日時 | 2022-10-25 16:39:28 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 508 ms / 2,000 ms |
コード長 | 2,494 bytes |
コンパイル時間 | 255 ms |
コンパイル使用メモリ | 86,960 KB |
実行使用メモリ | 83,932 KB |
最終ジャッジ日時 | 2023-09-16 14:41:52 |
合計ジャッジ時間 | 9,213 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 144 ms
78,852 KB |
testcase_01 | AC | 407 ms
83,544 KB |
testcase_02 | AC | 312 ms
81,028 KB |
testcase_03 | AC | 174 ms
79,184 KB |
testcase_04 | AC | 304 ms
81,372 KB |
testcase_05 | AC | 140 ms
78,944 KB |
testcase_06 | AC | 481 ms
83,128 KB |
testcase_07 | AC | 350 ms
83,932 KB |
testcase_08 | AC | 400 ms
83,584 KB |
testcase_09 | AC | 414 ms
83,320 KB |
testcase_10 | AC | 279 ms
81,108 KB |
testcase_11 | AC | 139 ms
77,968 KB |
testcase_12 | AC | 254 ms
80,012 KB |
testcase_13 | AC | 440 ms
83,560 KB |
testcase_14 | AC | 74 ms
71,352 KB |
testcase_15 | AC | 70 ms
71,392 KB |
testcase_16 | AC | 124 ms
77,544 KB |
testcase_17 | AC | 69 ms
71,284 KB |
testcase_18 | AC | 69 ms
71,056 KB |
testcase_19 | AC | 72 ms
71,252 KB |
testcase_20 | AC | 71 ms
71,312 KB |
testcase_21 | AC | 85 ms
76,364 KB |
testcase_22 | AC | 99 ms
76,600 KB |
testcase_23 | AC | 93 ms
76,648 KB |
testcase_24 | AC | 95 ms
76,544 KB |
testcase_25 | AC | 85 ms
76,372 KB |
testcase_26 | AC | 99 ms
76,548 KB |
testcase_27 | AC | 98 ms
76,440 KB |
testcase_28 | AC | 98 ms
76,556 KB |
testcase_29 | AC | 321 ms
82,564 KB |
testcase_30 | AC | 428 ms
83,552 KB |
testcase_31 | AC | 423 ms
83,388 KB |
testcase_32 | AC | 508 ms
82,920 KB |
ソースコード
class LazySegTree_RAQ: def __init__(self,init_val,segfunc,ide_ele): n = len(init_val) self.segfunc = segfunc self.ide_ele = ide_ele self.num = 1<<(n-1).bit_length() self.tree = [ide_ele]*2*self.num self.lazy = [0]*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 gindex(self,l,r): l += self.num r += self.num lm = l>>(l&-l).bit_length() rm = r>>(r&-r).bit_length() while r>l: if l<=lm: yield l if r<=rm: yield r r >>= 1 l >>= 1 while l: yield l l >>= 1 def propagates(self,*ids): for i in reversed(ids): v = self.lazy[i] if v==0: continue self.lazy[i] = 0 self.lazy[2*i] += v self.lazy[2*i+1] += v self.tree[2*i] += v self.tree[2*i+1] += v def add(self,l,r,x): ids = self.gindex(l,r) l += self.num r += self.num while l<r: if l&1: self.lazy[l] += x self.tree[l] += x l += 1 if r&1: self.lazy[r-1] += x self.tree[r-1] += x r >>= 1 l >>= 1 for i in ids: self.tree[i] = self.segfunc(self.tree[2*i], self.tree[2*i+1]) + self.lazy[i] def query(self,l,r): self.propagates(*self.gindex(l,r)) res = self.ide_ele l += self.num r += self.num while l<r: if l&1: res = self.segfunc(res,self.tree[l]) l += 1 if r&1: res = self.segfunc(res,self.tree[r-1]) l >>= 1 r >>= 1 return res #返す値 def segfunc(x, y): #return max(x, y) #return min(x, y) return (x+y)%(10**9+7) #return x*y #単位元 #ide_ele = -float('inf') #ide_ele = float('inf') ide_ele = 0 #ide_ele = 1 #update(l, r, x): 区間[l, r)の値にxを加算 #add(l, r): 区間[l, r)をsegfuncしたものを返す n,k=map(int,input().split()) dp=[0]*(n+1) dp[0]=1 seg = LazySegTree_RAQ(dp, segfunc, ide_ele) mod=10**9+7 for i in range(1,n+1): seg.add(i,min(i+k,n+1),seg.query(i-1,i)) print(seg.query(n,n+1))