結果
問題 | No.1887 K Consecutive Ks (Easy) |
ユーザー |
![]() |
提出日時 | 2022-03-25 22:59:34 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 2,055 bytes |
コンパイル時間 | 362 ms |
コンパイル使用メモリ | 81,792 KB |
実行使用メモリ | 87,040 KB |
最終ジャッジ日時 | 2024-10-14 07:04:25 |
合計ジャッジ時間 | 3,877 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 TLE * 1 |
other | -- * 16 |
ソースコード
import sys# sys.setrecursionlimit(200005)int1 = lambda x: int(x)-1pDB = lambda *x: print(*x, end="\n", file=sys.stderr)p2D = lambda x: print(*x, sep="\n", end="\n\n", file=sys.stderr)def II(): return int(sys.stdin.readline())def LI(): return list(map(int, sys.stdin.readline().split()))def LLI(rows_number): return [LI() for _ in range(rows_number)]def LI1(): return list(map(int1, sys.stdin.readline().split()))def LLI1(rows_number): return [LI1() for _ in range(rows_number)]def SI(): return sys.stdin.readline().rstrip()# dij = [(0, 1), (-1, 0), (0, -1), (1, 0)]# dij = [(0, 1), (-1, 0), (0, -1), (1, 0), (1, 1), (1, -1), (-1, 1), (-1, -1)]# inf = (1<<63)-1inf = (1 << 31)-1# md = 10**9+7md = 998244353class BitSum:def __init__(self, n):self.n = n+1self.table = [0]*self.ndef add(self, i, x):i += 1while i < self.n:self.table[i] += xself.table[i] %= mdi += i & -i# [0,i]の和def sum(self, i):i += 1res = 0while i > 0:res += self.table[i]res %= mdi -= i & -ireturn res# [l,r)の和def sumlr(self, l, r):if l >= r: return 0if l == 0: return self.sum(r-1)return (self.sum(r-1)-self.sum(l-1))%md# 数列を度数分布とみたときに、x番目がどのインデックスにあるかを返す# xが大きすぎるときは配列の長さnを返すdef rank(self, x):idx = 0d = 1 << (self.n-1).bit_length()-1while d:if idx+d < self.n and self.table[idx+d] < x:x -= self.table[idx+d]idx |= dd >>= 1return idxn, m = LI()bits = [BitSum(n+1) for _ in range(m+1)]bit = BitSum(n+1)bit.add(0, 1)for i in range(n):for a in range(1, m+1):val = bit.sumlr(i+2-a, i+1)-bits[a].sumlr(i+2-a, i+1)bits[a].add(i+1, val)bit.add(i+1, val)ans = pow(m, n, md)-bit.sumlr(n, n+1)print(ans%md)