結果

問題 No.950 行列累乗
ユーザー ttttan2ttttan2
提出日時 2019-12-13 02:21:30
言語 C++11
(gcc 11.4.0)
結果
WA  
実行時間 -
コード長 4,850 bytes
コンパイル時間 2,148 ms
コンパイル使用メモリ 192,072 KB
実行使用メモリ 32,256 KB
最終ジャッジ日時 2024-06-26 04:28:27
合計ジャッジ時間 17,794 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 AC 62 ms
12,160 KB
testcase_02 AC 84 ms
5,376 KB
testcase_03 AC 78 ms
5,376 KB
testcase_04 AC 1 ms
5,376 KB
testcase_05 AC 277 ms
32,128 KB
testcase_06 AC 283 ms
32,128 KB
testcase_07 AC 116 ms
5,376 KB
testcase_08 AC 288 ms
31,616 KB
testcase_09 AC 289 ms
32,128 KB
testcase_10 AC 139 ms
7,296 KB
testcase_11 AC 105 ms
5,376 KB
testcase_12 AC 87 ms
5,376 KB
testcase_13 AC 292 ms
32,000 KB
testcase_14 WA -
testcase_15 AC 100 ms
5,376 KB
testcase_16 AC 281 ms
26,368 KB
testcase_17 WA -
testcase_18 AC 95 ms
5,376 KB
testcase_19 AC 89 ms
5,376 KB
testcase_20 AC 90 ms
5,376 KB
testcase_21 AC 302 ms
32,128 KB
testcase_22 WA -
testcase_23 WA -
testcase_24 AC 280 ms
32,128 KB
testcase_25 WA -
testcase_26 WA -
testcase_27 AC 290 ms
32,128 KB
testcase_28 WA -
testcase_29 WA -
testcase_30 AC 292 ms
32,256 KB
testcase_31 WA -
testcase_32 WA -
testcase_33 AC 289 ms
32,000 KB
testcase_34 WA -
testcase_35 WA -
testcase_36 AC 176 ms
17,664 KB
testcase_37 WA -
testcase_38 WA -
testcase_39 AC 174 ms
17,664 KB
testcase_40 WA -
testcase_41 WA -
testcase_42 AC 277 ms
32,128 KB
testcase_43 WA -
testcase_44 WA -
testcase_45 AC 296 ms
32,128 KB
testcase_46 WA -
testcase_47 WA -
testcase_48 WA -
testcase_49 WA -
testcase_50 WA -
testcase_51 WA -
testcase_52 WA -
testcase_53 WA -
testcase_54 AC 2 ms
5,376 KB
testcase_55 AC 1 ms
5,376 KB
testcase_56 WA -
testcase_57 AC 280 ms
32,128 KB
testcase_58 WA -
testcase_59 WA -
testcase_60 AC 179 ms
17,920 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
//ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
typedef pair<pii,int> ppii;
typedef pair<int,pii> pipi;
typedef pair<ll,ll> pll;
typedef pair<pll,ll> ppll;
typedef pair<ll,pll> plpl;
typedef tuple<ll,ll,ll> tl;
ll mod;
ll mod2=998244353;
ll mod3=1000003;
ll mod4=998244853;
ll inf=1000000000;
double pi=2*acos(0);
#define rep(i,m,n) for(ll i=m;i<n;i++)
#define rrep(i,n,m) for(ll i=n;i>=m;i--)
int dh[4]={1,-1,0,0};
int dw[4]={0,0,1,-1};
int ddh[8]={-1,-1,-1,0,0,1,1,1};
int ddw[8]={-1,0,1,-1,1,-1,0,1};
ll lmax(ll a,ll b){
    if(a<b)return b;
    else return a;
}
ll lmin(ll a,ll b){
    if(a<b)return a;
    else return b;
}
ll gcd(ll a,ll b){
    if(a<b)swap(a,b);
    if(b==0)return a;
    if(a%b==0)return b;
    return gcd(b,a%b);
}
ll Pow(ll n,ll k){
    ll ret=1;
    ll now=n;
    while(k>0){
        if(k&1)ret*=now;
        now*=now;
        k/=2;
    }
    return ret;
}
ll beki(ll n,ll k,ll md){
  ll ret=1;
  ll now=n;
  while(k>0){
    if(k%2==1){
      ret*=now;
      ret%=md;
    }
    now*=now;
    now%=md;
    k/=2;
  }
  return ret;
}
ll gyaku(ll n,ll md){
  return beki(n,md-2,md);
}
vector<vector<ll>> matmul(vector<vector<ll>> a,vector<vector<ll>> b){
    ll n=a.size();
    vector<vector<ll>> ret;
    vector<ll> e;
    rep(i,0,n)e.push_back(0);
    rep(i,0,n)ret.push_back(e);
    rep(i,0,n){
        rep(j,0,n){
            ll sum=0;
            rep(k,0,n){
                sum+=(a[i][k]*b[k][j])%mod;
                sum%=mod;
            }
            ret[i][j]=sum;
        }
    }
    return ret;
}
vector<vector<ll>> matbeki(vector<vector<ll>> a,ll n){
    vector<vector<ll>> ret;
    ll m=a.size();
    vector<ll> e;
    rep(i,0,m)e.push_back(0);
    rep(i,0,m)ret.push_back(e);
    rep(i,0,m){
        rep(j,0,m){
            if(i==j)ret[i][j]=1;
            else ret[i][j]=0;
        }
    }
    
    while(n>0){
        if(n%2){
            ret=matmul(ret,a);
            
        }
        a=matmul(a,a);
        n/=2;
    }
    return ret;
}
int main(){
    ios::sync_with_stdio(false);cin.tie(0);
    
    ll p;cin>>p;
    mod=p;
    vector<vector<ll>> a,b;
    rep(i,0,2){
        vector<ll> v;
        rep(j,0,2){
            ll r;cin>>r;
            v.push_back(r);
        }
        a.push_back(v);
    }
    rep(i,0,2){
        vector<ll> v;
        rep(j,0,2){
            ll r;
            cin>>r;
            v.push_back(r);
        }
        b.push_back(v);
    }
    if(a[0][0]*a[1][1]-a[0][1]*a[1][0]!=0){
        ll u=a[0][0]*a[1][1]-a[0][1]*a[1][0];
        vector<vector<ll>> c;
        
        u=gyaku(u,mod);
        c={{(u*a[1][1])%mod,(u*(-a[0][1]+mod))%mod},
            {(u*(-a[1][0]+mod))%mod,(u*a[0][0])%mod}};
        ll m=1;
        while(m*m<5000000000)m++;
        map<vector<vector<ll>>,ll> mp;
        vector<vector<ll>> now={{1,0},{0,1}};
        rep(i,0,m){
            if(mp[now]==0)
            mp[now]=i+1;
            now=matmul(now,a);
        }
        vector<vector<ll>> g=b;
        c=matbeki(c,m);
        ll ans=inf;
        rep(i,0,m){
            if(mp[g]!=0){
                ans=min(ans,i*m+mp[g]-1);
            }
            g=matmul(g,c);
        }
        if(ans==inf)cout<<-1<<endl;
        else cout<<ans<<endl;
    }
    else{
        if(a[0][0]+a[1][1]==0){
            vector<vector<ll>> k={{0,0},{0,0}};
            if(b==k){
                if(a==k)cout<<1<<endl;
                else cout<<2<<endl;
                return 0;
            }
            if(a==b){
                cout<<1<<endl;
                return 0;
            }
            cout<<-1<<endl;
            return 0;
        }
        ll m=1;
        while(m*m<5000000000)m++;
        map<ll,ll> mp;
        ll now=1;
        rep(i,0,m){
            mp[now]=i+1;
            now*=(a[0][0]+a[1][1]);
            now%=mod;
        }
        ll h=-1,w=-1;
        rep(i,0,2){
            rep(j,0,2){
                if(a[i][j]!=0){
                    h=i;
                    w=j;
                    break;
                }
            }
            if(h>=0)break;
        }
        if(h==-1&&w==-1){
            if(a==b){
                cout<<1<<endl;
                return 0;
            }
        }
        ll g=b[h][w]*gyaku(a[h][w],mod);
        g%=mod;
        vector<ll> v;
        ll uu=gyaku(a[0][0]+a[1][1],mod);
        uu=beki(uu,m,mod);
        rep(i,0,m){
            if(mp[g]!=0){
                v.push_back(i*m+mp[g]-1);
            }
            g=g*uu;
            g%=mod;
        }
        rep(i,0,v.size()){
            vector<vector<ll>> f=a;
            f=matbeki(f,v[i]);
            if(f==b){
                cout<<v[i]<<endl;
                return 0;
            }
        }
        cout<<-1<<endl;
        return 0;
    }
}
0