#445492 (Python3) No.962 LCPs

提出ソース

結果

問題	No.962 LCPs
ユーザー	maspy
提出日時	2020-03-18 21:45:20
言語	Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果	AC
実行時間	535 ms / 2,000 ms
コード長	827 bytes
コンパイル時間	203 ms
コンパイル使用メモリ	12,800 KB
実行使用メモリ	51,476 KB
最終ジャッジ日時	2024-05-08 02:42:13
合計ジャッジ時間	10,729 ms
ジャッジサーバーID （参考情報）	judge5 / judge2

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	29 ms 10,880 KB
testcase_01	AC	29 ms 10,752 KB
testcase_02	AC	31 ms 10,880 KB
testcase_03	AC	30 ms 10,880 KB
testcase_04	AC	30 ms 10,752 KB
testcase_05	AC	29 ms 10,752 KB
testcase_06	AC	31 ms 10,752 KB
testcase_07	AC	29 ms 10,624 KB
testcase_08	AC	32 ms 10,752 KB
testcase_09	AC	30 ms 10,752 KB
testcase_10	AC	31 ms 10,752 KB
testcase_11	AC	30 ms 10,752 KB
testcase_12	AC	30 ms 10,752 KB
testcase_13	AC	29 ms 10,752 KB
testcase_14	AC	29 ms 10,752 KB
testcase_15	AC	29 ms 10,880 KB
testcase_16	AC	30 ms 10,880 KB
testcase_17	AC	535 ms 51,476 KB
testcase_18	AC	335 ms 35,736 KB
testcase_19	AC	334 ms 35,736 KB
testcase_20	AC	242 ms 27,508 KB
testcase_21	AC	236 ms 27,376 KB
testcase_22	AC	204 ms 23,384 KB
testcase_23	AC	201 ms 23,384 KB
testcase_24	AC	181 ms 20,756 KB
testcase_25	AC	176 ms 20,760 KB
testcase_26	AC	169 ms 18,936 KB
testcase_27	AC	211 ms 24,816 KB
testcase_28	AC	168 ms 19,432 KB
testcase_29	AC	154 ms 18,292 KB
testcase_30	AC	197 ms 22,860 KB
testcase_31	AC	235 ms 25,520 KB
testcase_32	AC	149 ms 17,576 KB
testcase_33	AC	308 ms 34,704 KB
testcase_34	AC	167 ms 19,612 KB
testcase_35	AC	161 ms 19,172 KB
testcase_36	AC	192 ms 20,676 KB
testcase_37	AC	113 ms 15,612 KB
testcase_38	AC	115 ms 15,604 KB
testcase_39	AC	118 ms 15,484 KB
testcase_40	AC	112 ms 15,480 KB
testcase_41	AC	115 ms 15,476 KB
testcase_42	AC	116 ms 15,736 KB
testcase_43	AC	113 ms 15,484 KB
testcase_44	AC	114 ms 15,552 KB
testcase_45	AC	115 ms 15,480 KB
testcase_46	AC	117 ms 15,348 KB
testcase_47	AC	30 ms 11,008 KB
testcase_48	AC	30 ms 11,008 KB
testcase_49	AC	29 ms 11,264 KB
testcase_50	AC	29 ms 11,008 KB
testcase_51	AC	29 ms 11,008 KB
testcase_52	AC	29 ms 11,136 KB
testcase_53	AC	29 ms 11,264 KB
testcase_54	AC	29 ms 11,008 KB
testcase_55	AC	29 ms 11,008 KB
testcase_56	AC	29 ms 11,136 KB
testcase_57	AC	80 ms 11,264 KB
testcase_58	AC	84 ms 11,008 KB
testcase_59	AC	80 ms 11,264 KB
testcase_60	AC	93 ms 11,264 KB
testcase_61	AC	94 ms 11,136 KB
testcase_62	AC	89 ms 11,136 KB
testcase_63	AC	338 ms 37,484 KB
testcase_64	AC	30 ms 11,264 KB
testcase_65	AC	339 ms 37,356 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#!/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)

yukicoder

結果

テストケース

ソースコード