結果

問題 No.962 LCPs
コンテスト
ユーザー H3PO4
提出日時 2021-07-25 16:27:48
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 580 ms / 2,000 ms
コード長 764 bytes
コンパイル時間 307 ms
コンパイル使用メモリ 12,416 KB
実行使用メモリ 76,868 KB
最終ジャッジ日時 2024-07-21 00:41:08
合計ジャッジ時間 19,466 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 64
権限があれば一括ダウンロードができます

ソースコード

diff # 

from dataclasses import dataclass, field

N = int(input())
ans = 0

tri = lambda x: x * (x + 1) // 2


@dataclass
class TrieNode:
    num: int
    last: int
    seq: int
    child: dict = field(default_factory=dict)


G = [None] * (2 * 10 ** 5 + 1)
G[0] = TrieNode(num=0, last=0, seq=0)

nodenum = 1
for i in range(N):
    S = input()
    pointer = 0
    for s in S:
        if s not in G[pointer].child:
            G[pointer].child[s] = nodenum
            G[nodenum] = TrieNode(num=nodenum, last=i - 1, seq=0)
            nodenum += 1
        pointer = G[pointer].child[s]
        if G[pointer].last == i - 1:
            G[pointer].seq += 1
        else:
            G[pointer].seq = 1
        ans += tri(G[pointer].seq)
        G[pointer].last = i
print(ans)
0