結果

問題 No.992 最長増加部分列の数え上げ
ユーザー noriocnorioc
提出日時 2024-07-17 05:42:54
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,428 ms / 2,000 ms
コード長 1,980 bytes
コンパイル時間 163 ms
コンパイル使用メモリ 82,760 KB
実行使用メモリ 271,744 KB
最終ジャッジ日時 2024-07-17 05:43:28
合計ジャッジ時間 30,824 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 42 ms
54,444 KB
testcase_01 AC 42 ms
55,128 KB
testcase_02 AC 42 ms
54,044 KB
testcase_03 AC 42 ms
56,352 KB
testcase_04 AC 553 ms
207,640 KB
testcase_05 AC 399 ms
167,728 KB
testcase_06 AC 690 ms
262,560 KB
testcase_07 AC 502 ms
210,284 KB
testcase_08 AC 277 ms
132,744 KB
testcase_09 AC 507 ms
185,676 KB
testcase_10 AC 670 ms
260,604 KB
testcase_11 AC 842 ms
262,460 KB
testcase_12 AC 201 ms
120,124 KB
testcase_13 AC 458 ms
201,404 KB
testcase_14 AC 475 ms
185,232 KB
testcase_15 AC 232 ms
121,260 KB
testcase_16 AC 1,229 ms
269,976 KB
testcase_17 AC 290 ms
136,024 KB
testcase_18 AC 440 ms
177,052 KB
testcase_19 AC 837 ms
263,216 KB
testcase_20 AC 1,418 ms
271,744 KB
testcase_21 AC 1,372 ms
271,708 KB
testcase_22 AC 1,355 ms
270,268 KB
testcase_23 AC 1,428 ms
270,072 KB
testcase_24 AC 1,379 ms
269,624 KB
testcase_25 AC 1,398 ms
270,204 KB
testcase_26 AC 1,372 ms
269,876 KB
testcase_27 AC 1,418 ms
270,068 KB
testcase_28 AC 1,405 ms
271,608 KB
testcase_29 AC 1,364 ms
269,688 KB
testcase_30 AC 358 ms
166,504 KB
testcase_31 AC 359 ms
166,376 KB
testcase_32 AC 350 ms
166,664 KB
testcase_33 AC 385 ms
166,108 KB
testcase_34 AC 357 ms
166,224 KB
testcase_35 AC 326 ms
164,868 KB
testcase_36 AC 326 ms
164,620 KB
testcase_37 AC 354 ms
164,852 KB
testcase_38 AC 324 ms
164,748 KB
testcase_39 AC 328 ms
164,772 KB
testcase_40 AC 441 ms
163,808 KB
testcase_41 AC 436 ms
163,680 KB
testcase_42 AC 407 ms
163,992 KB
testcase_43 AC 445 ms
163,516 KB
testcase_44 AC 426 ms
163,844 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from bisect import bisect_left
from collections import defaultdict


def compress(a: list) -> list:
    d = {v: i for i, v in enumerate(sorted(set(a)))}
    return [d[v] for v in a]


class FenwickTree:
    def __init__(self, n):
        self.data = [0] * (n+10)
        self.n = (n+10)

    def add(self, p, x):
        assert 0 <= p < self.n
        p += 1
        while p < len(self.data):
            self.data[p] += x
            p += p & -p

    def sum(self, p):
        """区間 [0, p] の和"""
        assert 0 <= p < self.n
        p += 1
        s = 0
        while p > 0:
            s += self.data[p]
            p -= p & -p
        return s

    def rangesum(self, l, r):
        """区間 [l, r] の和"""
        assert 0 <= l <= r < self.n
        s = self.sum(r)
        if l > 0:
            s -= self.sum(l-1)
        return s


# LIS のアルゴリズム
# 各要素のランクを返す
def lis(a: list) -> list:
    n = len(a)
    ranks = [0] * n  # ranks[i] : A[i] が LIS の何番目か
    dp = [INF] * n
    for i in range(n):
        ranks[i] = bisect_left(dp, a[i])
        dp[ranks[i]] = a[i]

    return ranks


MOD = 10 ** 9 + 7
INF = 1 << 60
N = int(input())
A = list(map(int, input().split()))
A = compress(A)

ranks = lis(A)
max_rank = max(ranks)
d = defaultdict(list)
for i in range(N):
    d[ranks[i]].append(i)

ft = FenwickTree(N)
dp = [0] * N
for p in d[max_rank]:
    ft.add(p, 1)
    dp[p] = 1

# print(f'{A=}')
# print(f'{ranks=}')
for r in reversed(range(max_rank)):
    pp = [0] * N
    dp, pp = pp, dp

    ft2 = FenwickTree(N)
    t = FenwickTree(N)
    up = d[r+1].copy()
    for p in reversed(d[r]):
        while up and p < up[-1]:
            q = up[-1]
            ft2.add(A[q], pp[q])
            up.pop()

        # print(f'{r=} {p=} {ft2.rangesum(A[p]+1, N)=}')
        t.add(A[p], ft2.rangesum(A[p]+1, N) % MOD)
        dp[p] = ft2.rangesum(A[p]+1, N) % MOD

    ft = t

# ans = ft.sum(N)
ans = sum(dp) % MOD
print(ans)
0