結果
問題 |
No.3231 2×2行列相似判定 〜hard〜
|
ユーザー |
👑 |
提出日時 | 2025-07-27 17:54:27 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 42 ms / 2,000 ms |
コード長 | 3,527 bytes |
コンパイル時間 | 360 ms |
コンパイル使用メモリ | 82,620 KB |
実行使用メモリ | 55,148 KB |
最終ジャッジ日時 | 2025-07-29 02:25:46 |
合計ジャッジ時間 | 3,007 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 37 |
ソースコード
class ModB: B = 10**9+7 length_bound = 10**6 #User definition length_max = min( length_bound , B ) inverse = [None,1] factorial = [1] factorial_inverse = [1] def SetModulo(B): ModB.B = int(B) assert(ModB.B > 0) ModB.length_max = min( ModB.length_bound , ModB.B ) ModB.inverse = [None,1] if ModB.B>1 else [0] ModB.factorial = [1 if ModB.B>1 else 0] ModB.factorial_inverse = [1 if ModB.B>1 else 0] def __init__(self,val,valid = False): self.val = int(val) if not valid and not(0 <= self.val < ModB.B):self.val %= ModB.B def copy(self): return ModB(self.val,True) def __str__(self): return str(self.val) def __eq__(self,x): if x.__class__ != __class__:x=ModB(x) return x.val==self.val def __ne__(self,other): return not( self == other ) def __iadd__(self,x): self.val += ModB.ref(x).val if self.val >= ModB.B:self.val -= ModB.B return self def __add__(self,x): a = self.copy() a += x return a def __radd__(self,x): return ModB(x + self.val) def __neg__(self): return ModB(ModB.B - self.val if self.val else 0,True) def __isub__(self,x): self.val -= ModB.ref(x).val if self.val < 0:self.val += ModB.B return self def __sub__(self,x): a = self.copy() a -= x return a def __rsub__(self,x): return ModB(x - self.val) def __mul__(self,x): return ModB(self.val * ModB.get(x)) def __imul__(self,x): self = self * x return self def __rmul__(self,x): return ModB(x * self.val) def __pow__(self,n): #Supported only if n>=0. assert 0<=n answer = ModB(1) power = self.copy() while n > 0: if n&1:answer *= power.val power *= power.val n >>= 1 return answer def __xor__(self,n): #Supported only if B is a prime and val!=0, or n>=0. return self ** ( ( n * (2 - ModB.B) )if n < 0 else n ) def Inverse(n): #Supported only if B is a prime. if n.__class__ == __class__:n=n.val if n >= ModB.B:n %= ModB.B assert n > 0 or ModB.B == 1 if n < ModB.length_max: while len(ModB.inverse) <= n:ModB.inverse+=[ModB.B - ModB.inverse[ModB.B % len(ModB.inverse)] * ( ModB.B // len(ModB.inverse) ) % ModB.B] return ModB(ModB.inverse[n],True) return ModB(n) ** ( ModB.B - 2 ) def __itruediv__(self,x): self *= ModB.Inverse(x) return self def __truediv__(self,x): return self * ModB.Inverse(x) def __rtruediv__(self,x): return x * ModB.Inverse(self.val) def Factorial(n): while len(ModB.factorial) <= n:ModB.factorial+=[ModB.factorial[-1] * len(ModB.factorial) % ModB.B] return ModB(ModB.factorial[n],True) def FactorialInverse(n): #Supported only if B is a prime. while len(ModB.factorial_inverse) <= n:ModB.factorial_inverse+=[ModB.factorial_inverse[-1] * ModB.Inverse( len(ModB.factorial_inverse) ).val % ModB.B] return ModB(ModB.factorial_inverse[n],True) def Combination(n,m): #Supported only if B is a prime. return ModB.Factorial(n) * (ModB.FactorialInverse(m).val * ModB.FactorialInverse(n-m).val)if 0<=m<=n else ModB(0,True) #private: def get(n): return n.val if n.__class__ == __class__ else n def ref(n): return n if n.__class__ == __class__ else ModB(n) def Similarity(A): zero=A[0][0].__class__(0) one=A[0][0].__class__(1) return[[one],[A[0][0]*A[1][1]-A[0][1]*A[1][0],-A[0][0]-A[1][1],one]]if A[0][1]!=0 or A[1][0]!=0 or A[0][0]!=A[1][1]else[[-A[0][0],one],[-A[1][1],one]] def Similar(A,B): return Similarity(A)==Similarity(B) R=range J=lambda:[[ModB(x,1)for x in map(int,input().split())]for i in R(2)] A=J() B=J() print("YNeos"[1^Similar(A,B)::2])