結果

問題 No.992 最長増加部分列の数え上げ
ユーザー noriocnorioc
提出日時 2024-07-17 05:40:50
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,944 bytes
コンパイル時間 151 ms
コンパイル使用メモリ 82,620 KB
実行使用メモリ 295,520 KB
最終ジャッジ日時 2024-07-17 05:41:14
合計ジャッジ時間 20,696 ms
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 43 ms
62,780 KB
testcase_01 AC 46 ms
55,712 KB
testcase_02 AC 45 ms
53,944 KB
testcase_03 AC 43 ms
55,120 KB
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 TLE -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 TLE -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
testcase_35 -- -
testcase_36 -- -
testcase_37 -- -
testcase_38 -- -
testcase_39 -- -
testcase_40 -- -
testcase_41 -- -
testcase_42 -- -
testcase_43 -- -
testcase_44 -- -
権限があれば一括ダウンロードができます

ソースコード

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


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))
        dp[p] = ft2.rangesum(A[p]+1, N)

    ft = t

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