結果

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

ソースコード

diff #

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