結果

問題 No.992 最長増加部分列の数え上げ
ユーザー noriocnorioc
提出日時 2024-07-17 23:25:15
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,056 ms / 2,000 ms
コード長 1,766 bytes
コンパイル時間 186 ms
コンパイル使用メモリ 82,160 KB
実行使用メモリ 271,476 KB
最終ジャッジ日時 2024-07-17 23:25:41
合計ジャッジ時間 25,834 ms
ジャッジサーバーID
(参考情報)
judge4 / judge2
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 47 ms
53,888 KB
testcase_01 AC 43 ms
53,760 KB
testcase_02 AC 43 ms
54,144 KB
testcase_03 AC 43 ms
53,888 KB
testcase_04 AC 447 ms
209,304 KB
testcase_05 AC 339 ms
170,344 KB
testcase_06 AC 514 ms
249,844 KB
testcase_07 AC 371 ms
209,792 KB
testcase_08 AC 227 ms
139,812 KB
testcase_09 AC 399 ms
194,616 KB
testcase_10 AC 526 ms
240,920 KB
testcase_11 AC 641 ms
261,580 KB
testcase_12 AC 194 ms
124,296 KB
testcase_13 AC 367 ms
190,556 KB
testcase_14 AC 368 ms
176,940 KB
testcase_15 AC 187 ms
120,508 KB
testcase_16 AC 891 ms
259,872 KB
testcase_17 AC 223 ms
141,732 KB
testcase_18 AC 373 ms
198,212 KB
testcase_19 AC 628 ms
261,416 KB
testcase_20 AC 1,011 ms
269,568 KB
testcase_21 AC 1,023 ms
270,016 KB
testcase_22 AC 1,023 ms
270,272 KB
testcase_23 AC 992 ms
271,476 KB
testcase_24 AC 1,021 ms
270,384 KB
testcase_25 AC 1,008 ms
270,076 KB
testcase_26 AC 1,056 ms
269,744 KB
testcase_27 AC 1,038 ms
269,784 KB
testcase_28 AC 1,014 ms
268,536 KB
testcase_29 AC 1,038 ms
269,680 KB
testcase_30 AC 305 ms
166,220 KB
testcase_31 AC 322 ms
166,272 KB
testcase_32 AC 300 ms
166,552 KB
testcase_33 AC 301 ms
165,864 KB
testcase_34 AC 298 ms
165,996 KB
testcase_35 AC 280 ms
164,904 KB
testcase_36 AC 273 ms
164,744 KB
testcase_37 AC 272 ms
164,864 KB
testcase_38 AC 270 ms
165,084 KB
testcase_39 AC 277 ms
164,720 KB
testcase_40 AC 350 ms
163,728 KB
testcase_41 AC 350 ms
164,012 KB
testcase_42 AC 347 ms
164,372 KB
testcase_43 AC 346 ms
163,628 KB
testcase_44 AC 342 ms
163,860 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_ranks(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 = compress(list(map(int, input().split())))

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

dp = [0] * N
for p in d[max_rank]:
    dp[p] = 1

for rank in reversed(range(max_rank)):
    pp = [0] * N
    dp, pp = pp, dp

    ft = FenwickTree(N)
    inds = d[rank+1].copy()
    for p in reversed(d[rank]):
        while inds and p < inds[-1]:
            k = inds[-1]
            ft.add(A[k], pp[k])
            inds.pop()

        dp[p] = ft.rangesum(A[p]+1, N) % MOD

ans = sum(dp) % MOD
print(ans)
0