#1017858 (PyPy3) No.2924 <===Super Spaceship String===>

• 問題: No.2924 <===Super Spaceship String===>
• ユーザー: prin_kemkem
• 提出日時: 2024-10-12 15:33:36
• 言語: PyPy3 (7.3.15)
• 結果: AC
• 実行時間: 110 ms / 2,000 ms
• コード長: 1,522 bytes
• コンパイル時間: 233 ms
• コンパイル使用メモリ: 82,048 KB
• 実行使用メモリ: 113,572 KB
• 最終ジャッジ日時: 2024-10-16 17:54:35
• 合計ジャッジ時間: 2,048 ms
• ジャッジサーバーID （参考情報）: judge3 / judge5

ファイルパターン	結果
sample	AC * 2
other	AC * 12

ソースコード

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
from collections import defaultdict, deque, Counter
# from functools import cache
# import copy
from itertools import combinations, permutations, product, accumulate, groupby, chain
# from more_itertools import distinct_permutations
from heapq import heapify, heappop, heappush
import math
import bisect
# from pprint import pprint
from random import randint, shuffle, randrange
# from sortedcontainers import SortedSet, SortedList, SortedDict
import sys
# sys.setrecursionlimit(2000000)
input = lambda: sys.stdin.readline().rstrip('\n')
inf = float('inf')
mod1 = 10**9+7
mod2 = 998244353
def ceil_div(x, y): return -(-x//y)
#################################################   
S = input()
stack = []
ans = 0
q = 0
for s in S:
    stack.append(s)
    if q == 0:
        if s == "<":
            q = 1
        else:
            ans += len(stack)
            stack.clear()
            q = 0
    elif q == 1:
        if s == "=":
            q = 2
        elif s == "<":
            q = 1
        else:
            ans += len(stack)
            stack.clear()
            q = 0
    else:
        if s == "=":
            q = 2
        elif s == ">":
            stack.pop()
            while stack[-1] == "=":
                stack.pop()
            stack.pop()
            if stack:
                if stack[-1] == "=":
                    q = 2
                elif stack[-1] == "<":
                    q = 1
            else:
                q = 0
        elif s == "<":
            q = 1
ans += len(stack)
print(ans)
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX