結果

問題 No.1892 Extended Fib Series
ユーザー 👑 p-adicp-adic
提出日時 2024-08-11 22:56:48
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 174 ms / 2,000 ms
コード長 1,382 bytes
コンパイル時間 450 ms
コンパイル使用メモリ 82,228 KB
実行使用メモリ 78,004 KB
最終ジャッジ日時 2024-08-11 22:56:53
合計ジャッジ時間 4,729 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 69 ms
75,940 KB
testcase_01 AC 132 ms
77,616 KB
testcase_02 AC 109 ms
76,964 KB
testcase_03 AC 76 ms
76,408 KB
testcase_04 AC 109 ms
76,928 KB
testcase_05 AC 68 ms
76,284 KB
testcase_06 AC 157 ms
77,888 KB
testcase_07 AC 129 ms
77,452 KB
testcase_08 AC 135 ms
77,740 KB
testcase_09 AC 139 ms
77,804 KB
testcase_10 AC 102 ms
77,148 KB
testcase_11 AC 68 ms
76,128 KB
testcase_12 AC 96 ms
76,656 KB
testcase_13 AC 132 ms
77,520 KB
testcase_14 AC 37 ms
53,404 KB
testcase_15 AC 37 ms
54,256 KB
testcase_16 AC 54 ms
67,792 KB
testcase_17 AC 37 ms
52,744 KB
testcase_18 AC 36 ms
54,044 KB
testcase_19 AC 37 ms
53,552 KB
testcase_20 AC 37 ms
52,828 KB
testcase_21 AC 45 ms
61,516 KB
testcase_22 AC 46 ms
60,564 KB
testcase_23 AC 44 ms
61,416 KB
testcase_24 AC 45 ms
62,348 KB
testcase_25 AC 44 ms
61,320 KB
testcase_26 AC 46 ms
62,432 KB
testcase_27 AC 45 ms
62,228 KB
testcase_28 AC 44 ms
61,488 KB
testcase_29 AC 174 ms
78,004 KB
testcase_30 AC 155 ms
77,812 KB
testcase_31 AC 152 ms
77,760 KB
testcase_32 AC 132 ms
77,832 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

R=range

def rec_str(a):
	if not isinstance(a,list):return str(a)
	return "".join(["[",", ".join(rec_str(x)for x in a),"]"])
class BIT:
	def __init__(self,x):
		if isinstance(x,int):
			self.N=x
			self.F=[0]*(x+1)
		else:
			self.N=len(x)
			self.F=[0]*(self.N+1)
			for j in R(1,self.N+1):
				i=j-1
				self.F[j]=x[i]
				k=j-(j&-j)
				while i>k:
					self.F[j]+=self.F[i]
					i-=i&-i
	def Add(self,i,u):
		i+=1
		while i<=self.N:
			self.F[i]+=u
			i+=i&-i
	def InitialSegmentSum(self,r):
		assert(r>-2)
		a=0
		i=min(r+1,self.N)
		while i:
			a+=self.F[i]
			i-=i&-i
		return a
class IntervalAddBIT:
	def __init__(self,N):
		self.N=N
		self.F=BIT(N+1)
		self.G=BIT(N+1)
	def IntervalAdd(self,l,r,u):
		self.F.Add(l,-(l-1)*u)
		self.F.Add(r+1,r*u)
		self.G.Add(l,u)
		self.G.Add(r+1,-u)
	def Add(self,i,u):
		self.IntervalAdd(i,i,u)
	def Set(self,i,u):
		self.Add(i,u-self.Get(i))
	def InitialSegmentSum(self,r):
		return self.F.InitialSegmentSum(r)+r*self.G.InitialSegmentSum(r)
	def IntervalSum(self,l,r):
		return self.InitialSegmentSum(r)-self.InitialSegmentSum(l-1)
	def Get(self,i):
		return self.IntervalSum(i,i)
	def list(self):
		return[self.Get(i)for i in R(self.N)]
	def __str__(self):
		return rec_str(self.list())

N,L=map(int,input().split())
P=10**9+7
bit=IntervalAddBIT(N+1)
bit.Add(0,1);
for i in R(N):bit.IntervalAdd(i+1,i+L,bit.Get(i)%P)
print(bit.Get(N)%P)
0