結果
問題 | No.2791 Beginner Contest |
ユーザー |
![]() |
提出日時 | 2024-06-21 21:47:20 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 591 ms / 2,000 ms |
コード長 | 5,188 bytes |
コンパイル時間 | 269 ms |
コンパイル使用メモリ | 82,496 KB |
実行使用メモリ | 95,812 KB |
最終ジャッジ日時 | 2024-06-21 21:47:27 |
合計ジャッジ時間 | 5,968 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 17 |
ソースコード
from collections import *import sysimport heapqimport bisectimport itertoolsfrom functools import lru_cachefrom types import GeneratorTypefrom fractions import Fractionimport mathimport copyimport random# sys.setrecursionlimit(int(1e7))# @lru_cache(maxsize=None) # CPython特化# @bootstrap # PyPy特化(こっちのほうが速い) yield dfs(), yield Noneを忘れずにdef bootstrap(f, stack=[]): # yielddef wrappedfunc(*args, **kwargs):if stack:return f(*args, **kwargs)else:to = f(*args, **kwargs)while True:if type(to) is GeneratorType:stack.append(to)to = next(to)else:stack.pop()if not stack:breakto = stack[-1].send(to)return toreturn wrappedfuncdxdy1 = ((0, 1), (0, -1), (1, 0), (-1, 0)) # 上下右左dxdy2 = ((0, 1),(0, -1),(1, 0),(-1, 0),(1, 1),(-1, -1),(1, -1),(-1, 1),) # 8方向すべてdxdy3 = ((0, 1), (1, 0)) # 右 or 下dxdy4 = ((1, 1), (1, -1), (-1, 1), (-1, -1)) # 斜めINF = float("inf")_INF = 1 << 60MOD = 998244353mod = 998244353MOD2 = 10**9 + 7mod2 = 10**9 + 7# memo : len([a,b,...,z])==26# memo : 2^20 >= 10^6# 小数の計算を避ける : x/y -> (x*big)//y ex:big=10**9# @:小さい文字, ~:大きい文字,None: 空の文字列# ユークリッドの互除法:gcd(x,y)=gcd(x,y-x)# memo : d 桁以下の p 進表記を用いると p^d-1 以下のすべての# 非負整数を表現することができる# memo : (X,Y) -> (X+Y,X−Y) <=> 点を原点を中心に45度回転し、√2倍に拡大# memo : (x,y)のx正から見た偏角をラジアンで(-πからπ]: math.atan2(y, x)# memo : a < bのとき ⌊a⌋ ≦ ⌊b⌋input = lambda: sys.stdin.readline().rstrip()mi = lambda: map(int, input().split())li = lambda: list(mi())ii = lambda: int(input())py = lambda: print("Yes")pn = lambda: print("No")pf = lambda: print("First")ps = lambda: print("Second")# ACLのではない。機能は減るが、こっちのほうが早いclass LazySegTree:def __init__(self, v, op, e, mapping, composition, id_):n = len(v)self._op = opself._e = eself._mapping = mappingself._composition = compositionself._id = id_self._size = 1 << (n - 1).bit_length()self._d = [e] * (2 * self._size)self._lz = [id_] * 2 * self._sizefor i in range(n):self._d[self._size + i] = v[i]for i in range(self._size - 1, 0, -1):self._d[i] = self._op(self._d[2 * i], self._d[2 * i + 1])def _gindex(self, l, r):l += self._sizer += self._sizelm = l >> (l & -l).bit_length()rm = r >> (r & -r).bit_length()while r > l:if l <= lm:yield lif r <= rm:yield rl >>= 1r >>= 1while l:yield ll >>= 1def _propagates(self, *ids):for i in reversed(ids):f = self._lz[i]self._lz[i] = self._idself._lz[2 * i] = self._composition(f, self._lz[2 * i])self._lz[2 * i + 1] = self._composition(f, self._lz[2 * i + 1])self._d[2 * i] = self._mapping(f, self._d[2 * i])self._d[2 * i + 1] = self._mapping(f, self._d[2 * i + 1])def apply(self, l, r, f):(*ids,) = self._gindex(l, r)self._propagates(*ids)l += self._sizer += self._sizewhile l < r:if l & 1:self._lz[l] = self._composition(f, self._lz[l])self._d[l] = self._mapping(f, self._d[l])l += 1if r & 1:self._lz[r - 1] = self._composition(f, self._lz[r - 1])self._d[r - 1] = self._mapping(f, self._d[r - 1])l >>= 1r >>= 1for i in ids:self._d[i] = self._op(self._d[2 * i], self._d[2 * i + 1])def prod(self, l, r):self._propagates(*self._gindex(l, r))resl = self._eresr = self._el += self._sizer += self._sizewhile l < r:if l & 1:resl = self._op(resl, self._d[l])l += 1if r & 1:resr = self._op(self._d[r - 1], resr)l >>= 1r >>= 1return self._op(resl, resr)# RMQのときdef op(a, b):return (a + b) % MODdef mapping(f, x): # 更新(dataに作用, fは上書きフラグ(f==ID: False))return (f + x) % MODdef composition(f, g): # 更新(lazyに作用, fは上書きフラグ(f==ID: False), 作用順はg→f, fは新しく入ってきたやつ)return (f + g) % MODini = 0ID = inie = iniN, K = mi()dp = LazySegTree([ini] * (N + 1), op, e, mapping, composition, ID)dp.apply(0, 1, 1)for i in range(N):if i + K <= N:dp.apply(i + K, N + 1, dp.prod(i, i + 1))print(dp.prod(0, N + 1))