結果
| 問題 |
No.1996 <><
|
| コンテスト | |
| ユーザー |
titia
|
| 提出日時 | 2022-07-02 01:33:21 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 870 ms / 2,000 ms |
| コード長 | 1,111 bytes |
| コンパイル時間 | 321 ms |
| コンパイル使用メモリ | 82,192 KB |
| 実行使用メモリ | 82,596 KB |
| 最終ジャッジ日時 | 2024-11-26 10:16:24 |
| 合計ジャッジ時間 | 10,892 ms |
|
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 4 |
| other | AC * 29 |
ソースコード
import sys
input = sys.stdin.readline
mod=10**9+7
N,K=map(int,input().split())
A=list(map(int,input().split()))
compression_dict={a: ind+1 for ind, a in enumerate(sorted(set(A)))}
A=[compression_dict[a] for a in A] # [2, 3, 0, 1, 4]
LEN=len(A)+5
def update(v,w): # index vにwを加える
while v<=LEN:
BIT[v]+=w
BIT[v]%=mod
v+=(v&(-v)) # v&(-v)で、最も下の立っているビット. 自分を含む大きなノードへ. たとえばv=3→v=4
def getvalue(v): # [1,v]の区間の和を求める
ANS=0
while v!=0:
ANS+=BIT[v]
ANS%=mod
v-=(v&(-v)) # 自分より小さい自分の和を構成するノードへ. たとえばv=14→v=12へ
return ANS
S=input().strip()
DP=[1]*N
for s in S:
BIT=[0]*(LEN+1)
NDP=[0]*N
if s=="<":
for i in range(N):
update(A[i],DP[i])
NDP[i]+=getvalue(A[i]-1)
NDP[i]%=mod
else:
for i in range(N):
update(A[i],DP[i])
NDP[i]+=getvalue(LEN)-getvalue(A[i])
NDP[i]%=mod
DP=NDP
print(sum(DP)%mod)
titia