結果

問題 No.2924 <===Super Spaceship String===>
コンテスト
ユーザー tnodino
提出日時 2024-09-27 21:55:15
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 708 ms / 2,000 ms
コード長 920 bytes
コンパイル時間 505 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 49,472 KB
最終ジャッジ日時 2024-10-16 17:52:57
合計ジャッジ時間 3,327 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

S = input()

T = []
cnt = 0
# 文字列を圧縮する
for i in range(len(S)):
    if S[i] == '=':
        cnt += 1
    else:
        if cnt > 0:
            T.append(('=', cnt))
            cnt = 0
        T.append((S[i], 1))
if cnt > 0:
    T.append(('=', cnt))

stc = []
# スタックに前から挿入する
for i in range(len(T)):
    # スタックの一番後ろの文字が = で、今見ている文字も = なら統合する
    if stc and stc[-1][0] == '=' and T[i][0] == '=':
        stc[-1] = (stc[-1][0], stc[-1][1] + T[i][1])
        continue
    stc.append(T[i])
    n = len(stc)
    if n < 3:
        continue
    # スタックの後ろ 3 文字が宇宙列ならスタックから取り除く
    if stc[n-3][0] == '<' and stc[n-2][0] == '=' and stc[n-1][0] == '>':
        for _ in range(3):
            stc.pop()

# スタックに残った長さの和を出力する
print(sum(x[1] for x in stc))
0