結果
| 問題 |
No.1996 <><
|
| コンテスト | |
| ユーザー |
MasKoaTS
|
| 提出日時 | 2024-12-01 16:46:36 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 785 ms / 2,000 ms |
| コード長 | 1,047 bytes |
| コンパイル時間 | 348 ms |
| コンパイル使用メモリ | 82,304 KB |
| 実行使用メモリ | 96,456 KB |
| 最終ジャッジ日時 | 2024-12-01 16:46:49 |
| 合計ジャッジ時間 | 10,899 ms |
|
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 4 |
| other | AC * 29 |
ソースコード
import sys
input = sys.stdin.readline
MOD = 10 ** 9 + 7
class BIT:
def __init__(self, N):
self.N = N
self.bit = [0] * (N + 1)
self.b = 1 << N.bit_length() - 1
def add(self, a, w):
x = a + 1
while(x <= self.N):
self.bit[x] += w
x += x & -x
def sum(self, a):
ret, x = 0, a
while(x > 0):
ret += self.bit[x]
x -= x & -x
return ret
def rsum(self, l, r):
return self.sum(r) - self.sum(l)
'''
Main Code
'''
N, K = map(int, input().split())
a = list(map(int, input().split()))
s = input()[:-1]
di = { x : i for i, x in enumerate(sorted(set(a))) }
for i, x in enumerate(a):
a[i] = di[x]
dp = [[0] * N for _ in [0] * (K + 1)]
dp[0] = [1] * N
for i, c in enumerate(s, 1):
bt = BIT(N)
for j, x in enumerate(a):
bt.add(x, dp[i - 1][j])
if(c == '<'):
dp[i][j] = bt.sum(x) % MOD
else:
dp[i][j] = bt.rsum(x + 1, N) % MOD
ans = sum(dp[K]) % MOD
print(ans)
MasKoaTS