結果
問題 | No.2101 [Cherry Alpha N] ずっとこの数列だったらいいのに |
ユーザー | roaris |
提出日時 | 2022-10-15 01:50:36 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 2,270 bytes |
コンパイル時間 | 238 ms |
コンパイル使用メモリ | 82,308 KB |
実行使用メモリ | 115,740 KB |
最終ジャッジ日時 | 2024-06-26 19:09:47 |
合計ジャッジ時間 | 62,426 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | WA | - |
testcase_01 | WA | - |
testcase_02 | WA | - |
testcase_03 | WA | - |
testcase_04 | WA | - |
testcase_05 | WA | - |
testcase_06 | WA | - |
testcase_07 | WA | - |
testcase_08 | WA | - |
testcase_09 | WA | - |
testcase_10 | WA | - |
testcase_11 | WA | - |
testcase_12 | WA | - |
testcase_13 | WA | - |
testcase_14 | WA | - |
testcase_15 | WA | - |
testcase_16 | WA | - |
testcase_17 | WA | - |
testcase_18 | WA | - |
testcase_19 | WA | - |
testcase_20 | WA | - |
testcase_21 | WA | - |
testcase_22 | WA | - |
testcase_23 | WA | - |
testcase_24 | WA | - |
testcase_25 | WA | - |
testcase_26 | WA | - |
testcase_27 | WA | - |
testcase_28 | WA | - |
testcase_29 | WA | - |
testcase_30 | WA | - |
testcase_31 | WA | - |
testcase_32 | WA | - |
testcase_33 | WA | - |
testcase_34 | WA | - |
testcase_35 | WA | - |
testcase_36 | WA | - |
testcase_37 | WA | - |
testcase_38 | WA | - |
testcase_39 | WA | - |
testcase_40 | WA | - |
testcase_41 | WA | - |
ソースコード
import sys input = sys.stdin.readline from collections import * import random class RollingHash1: def __init__(self, s): self.base = base1 self.mod = 10**9+7 self.acc = [0] self.power = [1] for i in range(len(s)): self.acc.append((self.acc[-1]*self.base%self.mod+(ord(s[i])-ord('a')))%self.mod) self.power.append(self.power[-1]*self.base%self.mod) def get(self, l, r): return (self.acc[r]-self.acc[l]*self.power[r-l])%self.mod class RollingHash2: def __init__(self, s): self.base = base2 self.mod = 10**9+9 self.acc = [0] self.power = [1] for i in range(len(s)): self.acc.append((self.acc[-1]*self.base%self.mod+(ord(s[i])-ord('a')))%self.mod) self.power.append(self.power[-1]*self.base%self.mod) def get(self, l, r): return (self.acc[r]-self.acc[l]*self.power[r-l])%self.mod N = int(input()) ws = set() base1 = random.randint(2, 10**9+6) base2 = random.randint(2, 10**9+8) for _ in range(N): S = input()[:-1] n = len(S) rh1 = RollingHash1(S) rh2 = RollingHash2(S) h1 = rh1.get(0, n) h2 = rh2.get(0, n) if (h1, h2) in ws: print('Yes') else: for i in range(n-1): nh1 = h1 nh1 -= (ord(S[i])-ord('a'))*rh1.power[n-1-i]%rh1.mod nh1 %= rh1.mod nh1 -= (ord(S[i+1])-ord('a'))*rh1.power[n-2-i]%rh1.mod nh1 %= rh1.mod nh1 += (ord(S[i+1])-ord('a'))*rh1.power[n-1-i]%rh1.mod nh1 %= rh1.mod nh1 += (ord(S[i])-ord('a'))*rh1.power[n-2-i]%rh1.mod nh1 %= rh1.mod nh2 = h2 nh2 -= (ord(S[i])-ord('a'))*rh2.power[n-1-i]%rh2.mod nh2 %= rh2.mod nh2 -= (ord(S[i+1])-ord('a'))*rh2.power[n-2-i]%rh2.mod nh2 %= rh2.mod nh2 += (ord(S[i+1])-ord('a'))*rh2.power[n-1-i]%rh2.mod nh2 %= rh2.mod nh2 += (ord(S[i])-ord('a'))*rh2.power[n-2-i]%rh2.mod nh2 %= rh2.mod if (nh1, nh2) in ws: print('Yes') break else: print('No') ws.add((h1, h2))