結果

問題 No.284 門松と魔法(2)
ユーザー rpy3cpp
提出日時 2015-09-30 00:51:54
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 2,042 bytes
コンパイル時間 129 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 61,484 KB
最終ジャッジ日時 2024-07-19 18:06:09
合計ジャッジ時間 16,575 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 29 WA * 1 TLE * 1 -- * 9
権限があれば一括ダウンロードができます

ソースコード

diff # 

import collections

def solve(N, As):
    As = compress(As)
    if len(As) < 3:
        return 0
    LH = [set([float('inf')])]
    LHprev = collections.defaultdict(int)
    LHrank = collections.defaultdict(int)
    LHprev[float('inf')] = -float('inf')
    LHrank[float('inf')] = 0
    
    HL = [set([-float('inf')])]
    HLprev = collections.defaultdict(int)
    HLrank = collections.defaultdict(int)
    HLprev[-float('inf')] = float('inf')
    HLrank[-float('inf')] = 0
    
    for a in As:
        update_data(a, True, LH, LHprev, HL, HLrank, HLprev)
        update_data(a, False, HL, HLprev, LH, LHrank, LHprev)
    
    return max(len(LH), len(HL)) - 1


def update_data(a, dLOW, LH, LHprev, HL, HLrank, HLprev):
    '''dLOW: True  -> a をHL に追加しようとしている。
             False -> a をLH に追加しようとそている。
       LH, HL 等は、dLOW = True の場合の変数名。呼び出し側で適宜入れ替える。
    '''
    LHmax = len(LH) - 1
    for k in range(LHmax, - 1, -1):
        if k + 1 < HLrank[a]:
            break
        for b in LH[k]:
            if (dLOW and b <= a) or ((not dLOW) and b >= a) or LHprev[b] == a:
                continue
            if HLrank[a] == k + 1:
                if HLprev[a] != b:
                    HLprev[a] = -1
                break
            else:
                HLrank[a] = k + 1
                HLprev[a] = b
                lenHL = len(HL)
                if k == lenHL:
                    HL.append(set())
                    HL.append(set([a]))
                elif k + 1 == lenHL:
                    HL.append(set([a]))
                elif k + 1 < lenHL:
                    HL[k + 1].add(a)

def compress(As):
    Bs = [As[0]]
    b = As[0]
    for a in As[1:]:
        if a != b:
            Bs.append(a)
            b = a
    return Bs

#import random
#N = 10000
#As = [random.randint(1, 100) for i in range(N)]

N = int(input())
As = list(map(int, input().split()))
ans = solve(N, As)
if ans < 3:
    print(0)
else:
    print(ans)
0