結果
| 問題 |
No.1996 <><
|
| コンテスト | |
| ユーザー |
titia
|
| 提出日時 | 2022-07-02 01:33:09 |
| 言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
| 結果 |
TLE
|
| 実行時間 | - |
| コード長 | 1,135 bytes |
| コンパイル時間 | 163 ms |
| コンパイル使用メモリ | 12,672 KB |
| 実行使用メモリ | 21,760 KB |
| 最終ジャッジ日時 | 2024-11-26 10:15:58 |
| 合計ジャッジ時間 | 48,233 ms |
|
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 4 |
| other | AC * 15 TLE * 14 |
ソースコード
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