結果

問題 No.992 最長増加部分列の数え上げ
ユーザー norioc
提出日時 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
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 1 WA * 11 TLE * 2 -- * 28
権限があれば一括ダウンロードができます

ソースコード

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