結果

問題 No.515 典型LCP
ユーザー sue_charosue_charo
提出日時 2017-05-06 00:51:32
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
RE  
実行時間 -
コード長 1,538 bytes
コンパイル時間 397 ms
コンパイル使用メモリ 10,976 KB
実行使用メモリ 38,332 KB
最終ジャッジ日時 2023-10-12 10:57:08
合計ジャッジ時間 2,848 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 RE -
testcase_01 RE -
testcase_02 RE -
testcase_03 AC 33 ms
10,180 KB
testcase_04 AC 32 ms
10,156 KB
testcase_05 RE -
testcase_06 RE -
testcase_07 RE -
testcase_08 RE -
testcase_09 RE -
testcase_10 RE -
testcase_11 RE -
testcase_12 RE -
testcase_13 RE -
testcase_14 RE -
testcase_15 RE -
testcase_16 RE -
権限があれば一括ダウンロードができます

ソースコード

diff # 

# coding: utf-8
import array, bisect, collections, copy, heapq, itertools, math, random, re, string, sys, time
sys.setrecursionlimit(10 ** 7)
INF = 10 ** 20
MOD = 10 ** 9 + 7


def II(): return int(input())
def ILI(): return list(map(int, input().split()))
def IAI(LINE): return [ILI() for __ in range(LINE)]
def IDI(): return {key: value for key, value in ILI()}


def makeQuery(N, M, x, d):
    query = []
    i = list(range(1, N + 1))
    j = list(range(1, N + 1))
    for k in range(1, M + 1):
        i[k] = (x // (N - 1)) + 1
        j[k] = (x % (N - 1)) + 1
        if i[k] > j[k]:
            i[k], j[k] = j[k], i[k]
        else:
            j[k] = j[k] + 1
        x = (x + d) % (N * (N - 1))
        query.append((i[k] - 1, j[k] - 1))

    return query


def makeLcp(N, s):
    lcp = [[0] * N for __ in range(N)]
    for i in range(N - 1):
        for j in range(i, N):
            s_i = s[i]
            s_j = s[j]
            n_len = min(len(s_i), len(s_j))
            n_lcp = 0
            for k in range(n_len):
                if s_i[k] == s_j[k]:
                    n_lcp += 1
                else:
                    break
            lcp[i][j] = n_lcp
            lcp[j][i] = n_lcp
    return lcp


def solve(N, s, M, x, d):
    query = makeQuery(N, M, x, d)
    lcp = makeLcp(N, s)
    ans = 0

    for i, j in query:
        ans += lcp[i][j]

    return ans


def main():
    N = II()
    s = [input() for __ in range(N)]
    M, x, d = ILI()
    print(solve(N, s, M, x, d))


if __name__ == "__main__":
    main()
0