結果

問題 No.1996 <><
ユーザー MasKoaTS
提出日時 2022-05-04 21:26:22
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 575 ms / 2,000 ms
コード長 1,005 bytes
コンパイル時間 286 ms
コンパイル使用メモリ 81,920 KB
実行使用メモリ 96,852 KB
最終ジャッジ日時 2024-11-26 02:54:45
合計ジャッジ時間 8,571 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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 = [1 if(c == '<') else 0 for c in 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 in range(1, K + 1):
	c = s[i - 1]
	bt = BIT(N)
	if(c == 1):
		for j, k in enumerate(a):
			bt.add(k, dp[i - 1][j])
			dp[i][j] = bt.sum(k) % mod
	else:
		for j, k in enumerate(b):
			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