結果
問題 |
No.1313 N言っちゃダメゲーム (4)
|
ユーザー |
![]() |
提出日時 | 2020-12-10 02:27:17 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 143 ms / 2,000 ms |
コード長 | 1,233 bytes |
コンパイル時間 | 192 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 91,520 KB |
最終ジャッジ日時 | 2024-09-19 09:02:31 |
合計ジャッジ時間 | 4,267 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 35 |
ソースコード
import sys input = sys.stdin.readline N, K = map(int, input().split()) S = list(input())[: -1] dp = [0] * (N + 1) class BIT: def __init__(self, n): self.n = n self.data = [0] * (n + 1) self.el = [0] * (n + 1) def sum(self, i): s = 0 while i > 0: s += self.data[i] i -= i & -i return s def add(self, i, x): self.el[i] += x while i <= self.n: self.data[i] += x i += i & -i def get(self, i, j = None): if j is None: return self.el[i] return self.sum(j) - self.sum(i) def lowerbound(self, s): x = 0 y = 0 for i in range(self.n.bit_length(), -1, -1): k = x + (1 << i) if k <= self.n and (y + self.data[k] < s): y += self.data[k] x += 1 << i return x + 1 fwk = BIT(N + 1) fwk.add(N, 1) dp[-1] = 1 if S[-1] == "x": dp[-2] = 1 fwk.add(N - 1, 1) for i in range(N - 2, -1, -1): if (i > 0) & (S[i - 1] == "x"): dp[i] = 1 fwk.add(i, 1) #print(i, S[i-1], dp[i]) continue r = min(N, i + K) s = fwk.get(i, r) if s < r - i: dp[i] = 1 if i: fwk.add(i, dp[i]) #print(i, r, s, S[i-1], dp[i]) #print(dp) f = 0 for i in range(1, K + 1): if dp[i] == 0: print(i) f = 1 if f == 0: print(0)