結果

問題 No.2365 Present of good number
ユーザー planesplanes
提出日時 2023-06-30 23:57:51
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 4,770 bytes
コンパイル時間 1,885 ms
コンパイル使用メモリ 178,032 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-07-07 11:40:20
合計ジャッジ時間 7,819 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 95 ms
5,248 KB
testcase_01 AC 96 ms
5,376 KB
testcase_02 WA -
testcase_03 WA -
testcase_04 AC 97 ms
5,376 KB
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 AC 96 ms
5,376 KB
testcase_17 AC 99 ms
5,376 KB
testcase_18 AC 97 ms
5,376 KB
testcase_19 AC 101 ms
5,376 KB
testcase_20 AC 97 ms
5,376 KB
testcase_21 AC 97 ms
5,376 KB
testcase_22 AC 101 ms
5,376 KB
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 AC 99 ms
5,376 KB
testcase_27 AC 95 ms
5,376 KB
testcase_28 AC 95 ms
5,376 KB
testcase_29 AC 98 ms
5,376 KB
testcase_30 AC 95 ms
5,376 KB
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 -
権限があれば一括ダウンロードができます

ソースコード

diff # 

  #include <bits/stdc++.h> 
using namespace std;
using ll =long long;
#define all(v) v.begin(),v.end()
 #define rep(i,a,b) for(int i=a;i<b;i++)
#define rrep(i,a,b) for(int i=a;i>=b;i--)

ll INF=2e18;

const ll mod = 1000000007;
class mint {
    long long x;
public:
    mint(long long x=0) : x((x%mod+mod)%mod) {}
    mint operator-() const { 
      return mint(-x);
    }
    mint& operator+=(const mint& a) {
        if ((x += a.x) >= mod) x -= mod;
        return *this;
    }
    mint& operator-=(const mint& a) {
        if ((x += mod-a.x) >= mod) x -= mod;
        return *this;
    }
    mint& operator*=(const  mint& a) {
        (x *= a.x) %= mod;
        return *this;
    }
    mint operator+(const mint& a) const {
        mint res(*this);
        return res+=a;
    }
    mint operator-(const mint& a) const {
        mint res(*this);
        return res-=a;
    }
    mint operator*(const mint& a) const {
        mint res(*this);
        return res*=a;
    }
    mint pow(ll t) const {
        if (!t) return 1;
        mint a = pow(t>>1);
        a *= a;
        if (t&1) a *= *this;
        return a;
    }
    // for prime mod
    mint inv() const {
        return pow(mod-2);
    }
    mint& operator/=(const mint& a) {
        return (*this) *= a.inv();
    }
    mint operator/(const mint& a) const {
        mint res(*this);
        return res/=a;
    }

    friend ostream& operator<<(ostream& os, const mint& m){
        os << m.x;
        return os;
    }
};




template <class T> 
struct Matrix {
    vector<vector<T>> A;
    
    Matrix() {}
    Matrix(size_t n) :A(n,vector<T> (n,0)) {}
    Matrix(size_t n,size_t m) :A(n,vector<T> (m,0)) {}
    

    size_t height() const {
        return (A.size());
    }

    size_t width() const {
        assert(height()!=0) ;
        return (A[0].size());
    }


  inline const vector< T > &operator[](int k) const {
    return (A.at(k));
  }
  
    inline vector<T> &operator[](int k) {
        return (A.at(k));
    }


void I() {
    assert(height()==width());
    for(int i=0;i<height();i++) A[i][i]=1;
    return ;
}



  Matrix &operator+=(const Matrix &B) {
    size_t n = height(), m = width();
    assert(n == B.height() && m == B.width());
    for(int i = 0; i < n; i++)
      for(int j = 0; j < m; j++)
        (*this)[i][j] += B[i][j];
    return (*this);
  }

  Matrix &operator-=(const Matrix &B) {
    size_t n = height(), m = width();
    assert(n == B.height() && m == B.width());
    for(int i = 0; i < n; i++)
      for(int j = 0; j < m; j++)
        (*this)[i][j] -= B[i][j];
    return (*this);
  }

  Matrix &operator*=(const Matrix &B) {
    size_t n = height(), m = B.width(), p = width();
    assert(p == B.height());
    vector< vector< T > > C(n, vector< T >(m, 0));
    for(int i = 0; i < n; i++)
      for(int j = 0; j < m; j++)
        for(int k = 0; k < p; k++)
          C[i][j] = (C[i][j] + (*this)[i][k] * B[k][j]);
    A.swap(C);
    return (*this);
  }

  Matrix &operator^=(long long k) {
    Matrix B = Matrix::I(height());
    while(k > 0) {
      if(k & 1) B *= *this;
      *this *= *this;
      k >>= 1LL;
    }
    A.swap(B.A);
    return (*this);
  }

Matrix &operator%=(const ll &B) {
      size_t n = height(), m = width();
    for(int i = 0; i < n; i++)
      for(int j = 0; j < m; j++)
        (*this)[i][j] %=B;
    return (*this);
  }

  Matrix operator+(const Matrix &B) const {
    return (Matrix(*this) += B);
  }

  Matrix operator-(const Matrix &B) const {
    return (Matrix(*this) -= B);
  }

  Matrix operator*(const Matrix &B) const {
    return (Matrix(*this) *= B);
  }

  Matrix operator^(const long long k) const {
    return (Matrix(*this) ^= k);
  }

};



ll mod_pow(ll x,ll n,ll mod) {
ll res=1;
  while(n>0) {
if(n&1) {
res=res*x%mod;
}
   x=x*x%mod;
    n>>=1;
  }
return res;
}
 


int main() {
    ios::sync_with_stdio(false);
  cin.tie(0);

ll N,K;cin>>N>>K;

ll ma=500;

vector<bool> note(ma+1);
vector<ll> p(0);

for(ll i=2;i<=ma;i++) {
  if(note[i]) continue;
  p.push_back(i);
  for(ll j=1;j*i<=ma;j++) {
    note[j*i]=true;
  }
}


ll n=N;
for(ll i=2;i<=sqrt(N);i++) {
  while(n%i==0) {
    n/=i;
  }
}

if(n>1&&p.back()<n) p.push_back(n);




ll k=(ll)p.size();

k=min(k,200LL);



Matrix<ll> mat(k);
Matrix<ll> a(k);




for(ll i=0;i<k;i++) {
  for(ll j=0;j<k;j++) {

    ll t=(p[j]+1);
    ll count=0;

    while(t%p[i]==0) {
      t/=p[i];
      count++;
    }

    mat[i][j]=count;

  }
}

a.I();






Matrix<ll> ans(k,1);
for(ll i=0;i<k;i++) {
  ll count=0;
  while(N%p[i]==0) {
    count++;
    N/=p[i];
  }
  ans[i][0]=count;
}


for(ll i=0;i<60;i++) {
  if(K&(1LL<<i)) {
    a*=mat;
  }
  mat*=mat;
}

a*=ans;

mint an=1;

for(ll i=0;i<k;i++) {
  an*=mod_pow(p[i],a[i][0],mod);
}

cout<<an<<endl;
}
0