結果

問題 No.515 典型LCP
コンテスト
ユーザー sue_charo
提出日時 2017-06-02 14:52:21
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 1,168 bytes
記録
コンパイル時間 123 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 288,128 KB
最終ジャッジ日時 2024-09-21 22:12:49
合計ジャッジ時間 5,180 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 2
other TLE * 1 -- * 14
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

# 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 read():
    N = II()
    s = [str(input()) for __ in range(N)]
    M, x, d = ILI()
    return (N, s, M, x, d)


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


def solve(N, s, M, x, d):
    query = makeQuery(N, M, x, d)
    ans = 0
    for q in query:
        s_1 = s[q[0]]
        s_2 = s[q[1]]
        count = 0
        for a, b in zip(s_1, s_2):
            if a != b:
                break
            count += 1
        ans += count
    return ans


def main():
    params = read()
    print(solve(*params))


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