結果

問題 No.2454 Former < Latter
ユーザー lam6er
提出日時 2025-03-20 20:42:29
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 159 ms / 2,000 ms
コード長 1,049 bytes
コンパイル時間 207 ms
コンパイル使用メモリ 82,824 KB
実行使用メモリ 110,984 KB
最終ジャッジ日時 2025-03-20 20:42:35
合計ジャッジ時間 3,552 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

def get_z(s):
    n = len(s)
    Z = [0] * n
    Z[0] = n
    l, r = 0, 0
    for i in range(1, n):
        if i <= r:
            Z[i] = min(r - i + 1, Z[i - l])
        while i + Z[i] < n and s[Z[i]] == s[i + Z[i]]:
            Z[i] += 1
        if i + Z[i] - 1 > r:
            l, r = i, i + Z[i] - 1
    return Z

def solve():
    import sys
    input = sys.stdin.read().split()
    idx = 0
    T = int(input[idx])
    idx += 1
    for _ in range(T):
        N = int(input[idx])
        idx += 1
        S = input[idx]
        idx += 1
        Z = get_z(S)
        count = 0
        for i in range(N - 1):
            start = i + 1
            len_u = i + 1
            len_v = N - start
            m = min(len_u, len_v)
            z_val = Z[start] if start < N else 0
            lcp = min(z_val, m)
            if lcp < m:
                if S[lcp] < S[start + lcp]:
                    count += 1
            else:
                if len_u < len_v:
                    count += 1
        print(count)

if __name__ == '__main__':
    solve()
0