結果
問題 | No.950 行列累乗 |
ユーザー | 37zigen |
提出日時 | 2019-12-13 01:59:06 |
言語 | Java21 (openjdk 21) |
結果 |
TLE
|
実行時間 | - |
コード長 | 3,439 bytes |
コンパイル時間 | 2,410 ms |
コンパイル使用メモリ | 79,956 KB |
実行使用メモリ | 127,864 KB |
最終ジャッジ日時 | 2024-06-26 03:52:01 |
合計ジャッジ時間 | 8,900 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | TLE | - |
testcase_01 | -- | - |
testcase_02 | -- | - |
testcase_03 | -- | - |
testcase_04 | -- | - |
testcase_05 | -- | - |
testcase_06 | -- | - |
testcase_07 | -- | - |
testcase_08 | -- | - |
testcase_09 | -- | - |
testcase_10 | -- | - |
testcase_11 | -- | - |
testcase_12 | -- | - |
testcase_13 | -- | - |
testcase_14 | -- | - |
testcase_15 | -- | - |
testcase_16 | -- | - |
testcase_17 | -- | - |
testcase_18 | -- | - |
testcase_19 | -- | - |
testcase_20 | -- | - |
testcase_21 | -- | - |
testcase_22 | -- | - |
testcase_23 | -- | - |
testcase_24 | -- | - |
testcase_25 | -- | - |
testcase_26 | -- | - |
testcase_27 | -- | - |
testcase_28 | -- | - |
testcase_29 | -- | - |
testcase_30 | -- | - |
testcase_31 | -- | - |
testcase_32 | -- | - |
testcase_33 | -- | - |
testcase_34 | -- | - |
testcase_35 | -- | - |
testcase_36 | -- | - |
testcase_37 | -- | - |
testcase_38 | -- | - |
testcase_39 | -- | - |
testcase_40 | -- | - |
testcase_41 | -- | - |
testcase_42 | -- | - |
testcase_43 | -- | - |
testcase_44 | -- | - |
testcase_45 | -- | - |
testcase_46 | -- | - |
testcase_47 | -- | - |
testcase_48 | -- | - |
testcase_49 | -- | - |
testcase_50 | -- | - |
testcase_51 | -- | - |
testcase_52 | -- | - |
testcase_53 | -- | - |
testcase_54 | -- | - |
testcase_55 | -- | - |
testcase_56 | -- | - |
testcase_57 | -- | - |
testcase_58 | -- | - |
testcase_59 | -- | - |
testcase_60 | -- | - |
ソースコード
import java.io.*; import java.util.*; class Main { public static void main(String[] args) { new Main().run(); } long MOD; long[][] pow(long[][] a, long n){ long[][] ret=new long[2][2]; ret[0][0]=ret[1][1]=1; for(;n>0;n>>=1,a=mul(a,a)){ if(n%2==1) ret=mul(ret,a); } return ret; } long[][] mul(long[][] a, long[][] b){ long[][] ret=new long[a.length][b[0].length]; for(int i=0;i<a.length;++i){ for(int j=0;j<b[i].length;++j){ for(int k=0;k<a[i].length;++k){ ret[i][j]+=a[i][k]*b[k][j]%MOD; } ret[i][j]%=MOD; } } return ret; } long det(long[][] a){ return (a[0][0]*a[1][1]%MOD-a[0][1]*a[1][0]%MOD+MOD)%MOD; } long[][] rndmat(long p){ long[][] ret=new long[2][2]; Random rnd=new Random(); for(int i=0;i<2;++i)for(int j=0;j<2;++j)ret[i][j]=rnd.nextInt((int)p); return ret; } void run() { /* long p=10; for(int i=0;i<1000;++i){ long[][] a=rndmat(p); long[][] b=rndmat(p); long ans0=exact(a,b,p); long ans1=solve(a,b,p); if(ans0!=ans1){ tr(ans0,ans1,a,b,p); } } */ /**/ Scanner sc=new Scanner(System.in); long p=sc.nextLong(); long[][] a=new long[2][2]; long[][] b=new long[2][2]; for(int i=0;i<2;++i)for(int j=0;j<2;++j)a[i][j]=sc.nextLong(); for(int i=0;i<2;++i)for(int j=0;j<2;++j)b[i][j]=sc.nextLong(); System.out.println(solve(a,b,p)); /**/ } long exact(long[][] a, long[][] b, long p){ MOD=p; for(int i=1;i<p*p;++i){ long[][] pw_a=pow(a,i); boolean equiv=true; for(int j=0;j<2;++j) for(int k=0;k<2;++k) equiv&=pw_a[j][k]==b[j][k]; if(equiv)return i; } return -1; } long solve(long[][] a, long[][] b, long p) { if(p==2)return exact(a,b,p); Scanner sc = new Scanner(System.in); MOD=p; long det0=det(a); long det1=det(b); long x=(det0==0&&det1==0?0:discretelog(det0,det1)); if(x==-1){ return -1; } long[][] na=pow(a,x); long ans=x; long ord=(det0==0?1:ord(det0,MOD)); long[][] c=pow(a,ord); for(int i=0;x+i<MOD*MOD;i+=ord){ boolean equiv=true; for(int j=0;j<2;++j) for(int k=0;k<2;++k) equiv&=na[j][k]==b[j][k]; if(equiv){ return x+i; } na=mul(na,c); } return -1; } long inv(long a){ return pow(a, MOD-2); } long ord(long a, long p) { if(a==1)return 1; long ret = p - 1; for (long div = 2; div * div <= p - 1; ++div) { if ((p - 1) % div != 0) continue; if (pow(a, div) == 1) ret = Math.min(ret, div); else if (pow(a, (p - 1) / div) == 1) ret = Math.min(ret, (p - 1) / div); } return ret; } long pow(long a, long n){ long ret=1; for(;n>0;n>>=1,a=a*a%MOD){ if(n%2==1)ret=ret*a%MOD; } return ret; } // return x s.t. a^x = b && x>0 long discretelog(long a, long b){ if(a==1){ if(b==1) return 1; else return -1; }else if(a==0){ if(b==0)return 1; else return -1; } // a^(um+v) = b // a^v = b a^(-m)^u int m=(int)Math.sqrt(MOD); long pw=1; HashMap<Long,Integer> map=new HashMap<>(); for(int v=0;v<=m;++v){ map.put(pw,v); pw=pw*a%MOD; } long ima=pow(inv(a),m); long ipw=1; for(int i=0;i<=m;++i){ if(map.containsKey(b*ipw)){ long ret=i*m+map.get(b*ipw); if(ret!=0)return ret; } ipw=ipw*ima%MOD; } return -1; } long gcd(long a,long b){ if(a>b)return gcd(b,a); if(a==0)return b; return gcd(a,b%a); } void tr(Object... objects) { System.out.println(Arrays.deepToString(objects)); } }