結果
問題 |
No.1859 ><<<
|
ユーザー |
👑 ![]() |
提出日時 | 2022-03-07 22:24:35 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 307 ms / 2,000 ms |
コード長 | 5,093 bytes |
コンパイル時間 | 720 ms |
コンパイル使用メモリ | 82,388 KB |
実行使用メモリ | 146,384 KB |
最終ジャッジ日時 | 2024-07-22 19:20:52 |
合計ジャッジ時間 | 14,078 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 41 |
ソースコード
class Rolling_Hash(): def __init__(self,S,base,mod): self.mod=mod self.base=base self.string=S self.power=power=[1]*(len(S)+1) L=len(S) self.hash=h=[0]*(L+1) v=0 for i in range(L): #h[i+1]=v=(v*base+ord(S[i]))%mod h[i+1]=v=(v*base+S[i])%mod v=1 for i in range(L): power[i+1]=v=(v*base)%mod def __hasher(self, X): assert len(X)<=len(self) h=0 for i in range(len(X)): h=(h*self.base+X[i])%self.mod return h def get(self,l,r): return (self.hash[r]-self.hash[l]*self.power[r-l])%self.mod def count(self,T,start=0): alpha=self.__hasher(T) K=0 for i in range(start,len(self)-len(T)+1): if alpha==self[i:i+len(T)]: K+=1 return K def find(self,T,start=0): alpha=self.__hasher(T) for i in range(start,len(self)-len(T)+1): if alpha==self[i:i+len(T)]: return i return -1 def rfind(self,T,start=0): alpha=self.__hasher(T) for i in range(len(self)-len(T),start-1,-1): if alpha==self[i:i+len(T)]: return i return -1 def index(self,T,start=0): alpha=self.__hasher(T) for i in range(start,len(self)-len(T)+1): if alpha==self[i:i+len(T)]: return i raise ValueError("substring not found") def __getitem__(self,index): if index.__class__==int: if index<0: index+=len(self.string) assert 0<=index<len(self.string) return self.get(index,index+1) elif index.__class__==slice: assert (index.step==None) or (index.step==1) L=index.start if index.start else 0 R=index.stop if index.stop else len(self) if L<0:L+=len(self) if R<0:R+=len(self) return self.get(L,R) def __len__(self): return len(self.string) #================================================= class Multi_Rolling_Hash(): def __init__(self, S, base, *mod): self.string=S self.__N=len(mod) self.__base=base self.__mod=mod self.rh=[Rolling_Hash(S,base,m) for m in mod] def get(self, l, r): return tuple(H[l:r] for H in self.rh) def __hasher(self, X): assert len(X)<=len(self) h=[0]*self.__N for i in range(self.__N): m=self.__mod[i] a=0 for x in X: a=(a*self.__base+x)%m h[i]=a return tuple(h) def __getitem__(self, index): if index.__class__==int: if index<0: index+=len(self.string) assert 0<=index<len(self.string) return tuple(H[index] for H in self.rh) elif index.__class__==slice: assert (index.step==None) or (index.step==1) L=index.start if index.start else 0 R=index.stop if index.stop else len(self) if L<0:L+=len(self) if R<0:R+=len(self) return tuple(H[L:R] for H in self.rh) def count(self,T,start=0): alpha=self.__hasher(T) K=0 T_len=len(T) for i in range(start,len(self)-len(T)+1): if alpha==self[i:i+T_len]: K+=1 return K def find(self,T,start=0): alpha=self.__hasher(T) for i in range(start,len(self)-len(T)+1): if alpha==self[i:i+len(T)]: return i return -1 def rfind(self,T,start=0): alpha=self.__hasher(T) for i in range(len(self)-len(T),start-1,-1): if alpha==self[i:i+len(T)]: return i return -1 def index(self,T,start=0): alpha=self.__hasher(T) for i in range(start,len(self)-len(T)+1): if alpha==self[i:i+len(T)]: return i raise ValueError("substring not found") def __len__(self): return len(self.string) class Hasher(): def __init__(self, length, base, mod): self.length=length self.base=base self.mod=mod self.power=pw=[1]*length for i in range(1,length): pw[i]=(base*pw[i-1])%mod def __repr__(self): return "length: {}\nbase: {}\nmod: {}".format(self.length, self.base, self.mod) def encode(self, S): code=0; N=len(S) for i in range(N): #code+=ord(S[i])*self.power[N-1-i]%self.mod code+=S[i]*self.power[N-1-i]%self.mod return code%self.mod def decode(self, S): pass #================================================== from random import randint as ri N=int(input()) A=list(map(int,input().split())) S=input() Mod1=10**9+7; Mod2=10**9+9 a=ri(2,Mod1-1); b=ri(2,Mod1-1); p=ri(2,Mod1-1) B=[a if A[i%N]<A[(i+1)%N] else b for i in range(2*N-1)] T=[a if S[i]=="<" else b for i in range(N-1)] H=Multi_Rolling_Hash(B,p,Mod1,Mod2) print(H.find(T))