結果

問題 No.962 LCPs
コンテスト
ユーザー mkawa2
提出日時 2019-12-25 20:51:59
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 1,319 bytes
コンパイル時間 136 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 44,416 KB
最終ジャッジ日時 2024-09-25 11:21:14
合計ジャッジ時間 5,465 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 15 TLE * 1 -- * 48
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

sys.setrecursionlimit(10 ** 6)
int1 = lambda x: int(x) - 1
p2D = lambda x: print(*x, sep="\n")
def MI(): return map(int, sys.stdin.readline().split())
def LI(): return list(map(int, sys.stdin.readline().split()))
def LLI(rows_number): return [LI() for _ in range(rows_number)]

class SparseTable:
    def __init__(self, aa):
        w = len(aa)
        h = w.bit_length()
        table = [aa] + [[""] * w for _ in range(h - 1)]
        tablei1 = table[0]
        for i in range(1, h):
            tablei = table[i]
            for j in range(w-(1<<i)+1):
                rj = j + (1 << (i - 1))
                tablei[j] = self.make_lcp(tablei1[j], tablei1[rj])
            tablei1 = tablei
        self.table = table
        #print(table)

    def make_lcp(self,s,t):
        res=""
        for c0,c1 in zip(s,t):
            if c0!=c1:break
            res+=c0
        return res

    def lcp(self, l, r):
        i = (r - l).bit_length() - 1
        tablei = self.table[i]
        Lmin = tablei[l]
        Rmin = tablei[r - (1 << i)]
        res=self.make_lcp(Lmin,Rmin)
        return res

def main():
    n=int(input())
    ss=[input() for _ in range(n)]
    sp=SparseTable(ss)
    ans=0
    for j in range(n+1):
        for i in range(j):
            ans+=len(sp.lcp(i,j))*(j-i)
    print(ans)

main()
0