Submission #71145482

---

Source Code Expand

import sys
# sys.setrecursionlimit(10 ** 6)

# import numpy as np
# <nonumba> from numba import njit, jit, boolean, int64, float64, typeof




def main():
    from io import StringIO
    execute(StringIO(sample), sys.stderr)
    print("-- ^sample! --", file=sys.stderr)

    execute(sys.stdin, sys.stdout)

    # #<nonumba> if len(solve.nopython_signatures) != 1:
    # #<nonumba>     raise KeyboardInterrupt("!! Unexpected: wrong use of jit !!")

def execute(src, dst):

    S = [int(c) for c in src.readline().strip()]
    ans = solve(S)
    print(ans, file=dst)

sample = """\
7788788
"""

# #<nonumba> @jit(nopython=True, cache=True)
def solve(S):
    ans = 0
    h = -4
    nh = 0
    t = -2
    nt = 0
    for c in S + [-6,]:
        if c != t :
            if t == h + 1 :
                ans += min (nh , nt)
            h, nh = t, nt
            t = c
            nt = 1
        else :
            nt += 1
    return ans



main()

Submission Info

Judge Result