結果

問題 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
権限があれば一括ダウンロードができます

ソースコード

diff # 

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)
0