結果
問題 |
No.962 LCPs
|
ユーザー |
![]() |
提出日時 | 2020-03-18 21:45:20 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 553 ms / 2,000 ms |
コード長 | 827 bytes |
コンパイル時間 | 193 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 51,480 KB |
最終ジャッジ日時 | 2024-12-14 02:31:16 |
合計ジャッジ時間 | 11,319 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 64 |
ソースコード
#!/usr/bin/env python3.8 import sys read = sys.stdin.buffer.read readline = sys.stdin.buffer.readline readlines = sys.stdin.buffer.readlines from operator import itemgetter N = int(readline()) S = read().split() answer = 0 for word in S: answer += len(word) LCP = [0] * (N - 1) for i, (w1, w2) in enumerate(zip(S, S[1:])): L1 = len(w1) L2 = len(w2) j = 0 while j < L1 and j < L2 and w1[j] == w2[j]: j += 1 LCP[i] = j INF = 10 ** 6 LCP.insert(0, -INF) LCP.append(-INF) values = sorted(enumerate(LCP), key=itemgetter(1), reverse=True) left = list(range(1, N + 2)) right = list(range(-1, N)) for i, value in values[:-2]: l = left[i - 1] r = right[i + 1] left[r] = l right[l] = r cnt = (i - l + 1) * (r - i + 1) answer += value * cnt * (r - l + 4) // 2 print(answer)