結果
| 問題 |
No.1996 <><
|
| コンテスト | |
| ユーザー |
MasKoaTS
|
| 提出日時 | 2022-05-05 20:13:24 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 605 ms / 2,000 ms |
| コード長 | 884 bytes |
| コンパイル時間 | 294 ms |
| コンパイル使用メモリ | 82,304 KB |
| 実行使用メモリ | 95,796 KB |
| 最終ジャッジ日時 | 2024-11-26 02:57:36 |
| 合計ジャッジ時間 | 9,053 ms |
|
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 4 |
| other | AC * 29 |
ソースコード
import sys
input = sys.stdin.readline
mod = 10 ** 9 + 7
class BIT:
def __init__(self, n):
self.size = n
self.bit = [0] * (n + 1)
def add(self, k, p):
t = k + 1
while(t <= self.size):
self.bit[t] += p
self.bit[t] %= mod
t += t & -t
def sum(self, r):
res, t = 0, r
while(t > 0):
res += self.bit[t]
res %= mod
t -= t & -t
return res
"""
Main Code
"""
N, K = map(int, input().split())
a = list(map(int, input().split()))
s = input().rstrip()
lis = sorted(set(a))
d = {}
for i, k in enumerate(lis):
d[k] = i
for i in range(N):
a[i] = d[a[i]]
b = [len(lis) - 1 - i for i in a]
dp = [[0] * N for _ in [0] * (K + 1)]
dp[0] = [1] * N
for i, c in enumerate(s, 1):
bt = BIT(N)
lis = a if(c == '<') else b
for j, k in enumerate(lis):
bt.add(k, dp[i - 1][j])
dp[i][j] = bt.sum(k) % mod
ans = 0
for t in dp[K]:
ans += t
ans %= mod
print(ans)
MasKoaTS