結果
問題 |
No.515 典型LCP
|
ユーザー |
|
提出日時 | 2020-09-20 17:49:54 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 716 ms / 1,000 ms |
コード長 | 1,458 bytes |
コンパイル時間 | 174 ms |
コンパイル使用メモリ | 81,928 KB |
実行使用メモリ | 131,048 KB |
最終ジャッジ日時 | 2024-06-24 10:51:58 |
合計ジャッジ時間 | 5,698 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 15 |
ソースコード
import sys input = lambda : sys.stdin.readline().rstrip() sys.setrecursionlimit(max(1000, 10**9)) write = lambda x: sys.stdout.write(x+"\n") class SparseTable: op = min def __init__(self, a): n = len(a) t = [] c = 1 l = a while True: t.append(l) c *= 2 if c>n: break v = c//2 l = [self.op(l[i], l[i+v]) for i in range(n-v)] + l[-v:] self.t = t def query(self, l, r): """a[l,r)の最小値 assert l<r and l,r<n """ # t[i][j]: a[j:j+2^i)の最小値 b = (r-l).bit_length()-1 return self.op(self.t[b][l], self.t[b][r-(1<<b)]) def sub(s,t): i = 0 m = min(len(s), len(t)) while i<m and s[i]==t[i]: i += 1 return i from operator import itemgetter n = int(input()) s = [input() for _ in range(n)] ss = sorted(list(enumerate(s)), key=itemgetter(1)) vals = [sub(ss[i][1], ss[i+1][1]) for i in range(n-1)] st = SparseTable(vals) rsa = [None]*n for i,item in enumerate(ss): rsa[item[0]] = i M,x,d = map(int, input().split()) ans = 0 def sub(i,j): ii,jj = rsa[i], rsa[j] if ii>jj: ii,jj = jj,ii return st.query(ii,jj) for k in range(1, M+1): i = (x // (n-1)) + 1 j = (x % (n-1)) + 1 if i>j: i,j = j,i else: j += 1 x = (x + d) % (n * (n-1)) # print(i,j) ans += sub(i-1,j-1) print(ans)