結果
問題 | No.1996 <>< |
ユーザー | ygd. |
提出日時 | 2022-07-01 23:14:34 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,707 ms / 2,000 ms |
コード長 | 2,855 bytes |
コンパイル時間 | 408 ms |
コンパイル使用メモリ | 82,316 KB |
実行使用メモリ | 125,872 KB |
最終ジャッジ日時 | 2024-05-04 17:44:31 |
合計ジャッジ時間 | 21,225 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 45 ms
52,352 KB |
testcase_01 | AC | 42 ms
52,224 KB |
testcase_02 | AC | 43 ms
52,864 KB |
testcase_03 | AC | 61 ms
64,640 KB |
testcase_04 | AC | 41 ms
52,224 KB |
testcase_05 | AC | 42 ms
52,352 KB |
testcase_06 | AC | 42 ms
52,480 KB |
testcase_07 | AC | 43 ms
52,736 KB |
testcase_08 | AC | 112 ms
76,096 KB |
testcase_09 | AC | 54 ms
64,896 KB |
testcase_10 | AC | 87 ms
76,160 KB |
testcase_11 | AC | 771 ms
101,248 KB |
testcase_12 | AC | 1,500 ms
125,184 KB |
testcase_13 | AC | 1,707 ms
125,872 KB |
testcase_14 | AC | 1,602 ms
124,048 KB |
testcase_15 | AC | 1,513 ms
119,336 KB |
testcase_16 | AC | 1,334 ms
114,176 KB |
testcase_17 | AC | 1,434 ms
120,044 KB |
testcase_18 | AC | 867 ms
104,116 KB |
testcase_19 | AC | 1,034 ms
106,376 KB |
testcase_20 | AC | 974 ms
104,240 KB |
testcase_21 | AC | 1,258 ms
115,840 KB |
testcase_22 | AC | 1,467 ms
118,844 KB |
testcase_23 | AC | 148 ms
79,488 KB |
testcase_24 | AC | 889 ms
106,840 KB |
testcase_25 | AC | 473 ms
88,000 KB |
testcase_26 | AC | 775 ms
100,224 KB |
testcase_27 | AC | 213 ms
82,176 KB |
testcase_28 | AC | 80 ms
72,576 KB |
testcase_29 | AC | 471 ms
85,376 KB |
testcase_30 | AC | 115 ms
77,532 KB |
testcase_31 | AC | 249 ms
83,200 KB |
testcase_32 | AC | 101 ms
76,364 KB |
ソースコード
import sys #input = sys.stdin.readline #input = sys.stdin.buffer.readline #文字列はダメ #sys.setrecursionlimit(1000000) #import bisect #import itertools #import random #from heapq import heapify, heappop, heappush #from collections import defaultdict #from collections import deque #import copy #import math #from functools import lru_cache #@lru_cache(maxsize=None) MOD = pow(10,9) + 7 #MOD = 998244353 #dx = [1,0,-1,0] #dy = [0,1,0,-1] #dx8 = [1,1,0,-1,-1,-1,0,1] #dy8 = [0,1,1,1,0,-1,-1,-1] #和のセグ木にするときは以下の関数を定義する必要あり。 def segfunc(x,y): return x+y class SegmentTree(object): def __init__(self, A, dot, unit): n = 1 << (len(A) - 1).bit_length() tree = [unit] * (2 * n) for i, v in enumerate(A): tree[i + n] = v for i in range(n - 1, 0, -1): tree[i] = dot(tree[i << 1], tree[i << 1 | 1]) self._n = n self._tree = tree self._dot = dot self._unit = unit def __getitem__(self, i): return self._tree[i + self._n] def update(self, i, v): i += self._n self._tree[i] = v while i != 1: i >>= 1 self._tree[i] = self._dot(self._tree[i << 1], self._tree[i << 1 | 1]) def add(self, i, v): self.update(i, self[i] + v) def sum(self, l, r): #これで[l,r)から取り出す。 l += self._n r += self._n l_val = r_val = self._unit while l < r: if l & 1: l_val = self._dot(l_val, self._tree[l]) l += 1 if r & 1: r -= 1 r_val = self._dot(self._tree[r], r_val) l >>= 1 r >>= 1 return self._dot(l_val, r_val) def compress(S): S = set(S); S = list(S) S.sort() dic_comp_ori = {} dic_ori_comp = {} for i,a in enumerate(S): dic_ori_comp[a] = i dic_comp_ori[i] = a return dic_comp_ori, dic_ori_comp def main(): N,K = map(int,input().split()) A = list(map(int,input().split())) dic_comp_ori, dic_ori_comp = compress(A) A = [dic_ori_comp[a] for a in A] MAX = len(dic_comp_ori) s = str(input()) # print(A) # dp[k][v]: 今k番目の不等号まで完了して、直前の値がvとなる場合の数 dp = [SegmentTree([0]*MAX,segfunc,0) for _ in range(K+1)] for a in A: # 初期条件 dp[0].add(a,1) for k in range(K): # a未満の和 if s[k] == '<': num = dp[k].sum(0,a) # aより大の和 else: num = dp[k].sum(a+1,MAX) temp = dp[k+1][a] + num temp %= MOD dp[k+1].update(a,temp) ans = dp[K].sum(0,MAX) print(ans%MOD) if __name__ == '__main__': main()