結果

問題 No.181 A↑↑N mod M
コンテスト
ユーザー 小指が強い人
提出日時 2016-08-24 21:00:18
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 3,252 bytes
コンパイル時間 2,185 ms
コンパイル使用メモリ 170,740 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-08 02:10:51
合計ジャッジ時間 3,491 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 6
other AC * 13 WA * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"

using namespace std;

typedef long long ll;
typedef unsigned long long ull;
typedef vector<int> veci;
typedef vector<ll> vecll;
typedef vector<string> vecs;
template<class T,class U> using Hash=unordered_map<T,U>;

#define REP(i, a, n) for(ll i = (a); (i) < (ll)(n); (i)++)
#define RREP(i, a, n) for(ll i = (n)-1; (i) >= (a); (i)--)
#define rep(i, n) REP(i, 0, n)
#define rrep(i, n) RREP(i, 0, n)
#define MD 1000000007ULL
#define _SPLIT " "

template<class T> T read(){T a;cin >> a;return a;}
template<class T> void read(T& a){cin >> a;}
template<class T,class ...Args> void read(T& a, Args&... args){cin >> a; read(args...);} 
template<class T> void rarr(T& a, int n){for(int i = 0; i < n; i++) {cin >> a[i];}}
template<class T> void write(T a){cout << setprecision(22) << a << endl;}
template<class T,class ...Args> void write(T a, Args... args){cout << setprecision(22) << a << _SPLIT; write(args...);}
template<class T> void warr(vector<T> a, const char* c = " "){cout << a[0];for(int i = 1; i < (int)a.size(); i++)cout << c << a[i];cout << endl;;}
template<class T> void warr(T a, int n, const char* c = " "){cout << a[0];for(int i = 1; i < n; i++)cout << c << a[i];cout << endl;}
void split(string s, string delim, veci& result){result.clear();string::size_type pos = 0;while(pos != string::npos){string::size_type p = s.find(delim, pos);if(p == string::npos){result.push_back(atoi(s.substr(pos).data()));break;}else {result.push_back(atoi(s.substr(pos, p - pos).data()));}pos = p + delim.size();}}
void split(string s, string delim, vecs& result){result.clear();string::size_type pos = 0;while(pos != string::npos){string::size_type p = s.find(delim, pos);if(p == string::npos){result.push_back(s.substr(pos));break;}else {result.push_back(s.substr(pos, p - pos));}pos = p + delim.size();}}
ll gcd(ll a, ll b){while(true){ll k = a % b;if(k == 0)return b;a = b;b = k;}}
ll comb(ll n, ll m){ll p=1;m=min(m,n-m);for(ll i=1;i<=m;i++){p*=n-i+1;p/=i;}return p;}

ull power( const ull& a, ull k, const ull& n )
{
  if ( a == 0 || n == 0 ) return( 0 );
  if ( k == 0 ) return( 1 % n );
  
  ull currentMod = a % n;
  ull currentValue = ( ( k & 1 ) > 0 ) ? currentMod : 1;

  for ( k >>= 1 ; k > 0 ; k >>= 1 ) {
    currentMod = ( currentMod * currentMod ) % n;
    if ( ( k & 1 ) > 0 )
      currentValue = ( currentValue * currentMod ) % n;
  }

  return( currentValue );
}

int main(void)
{
    set<int> st;
    vector<int> v1;
    ll a,n,m;
    read(a,n,m);
    if(n==0)write(1%m),exit(0);
    ll p=1;
    while(st.count(p)==0){
        st.insert(p);
        v1.push_back(p);
        p=(p*a)%m;
    }
    //warr(v1);
    ll s1=find(v1.begin(),v1.end(),p)-v1.begin();
    st.clear();
    ll vs1=v1.size();
    ll d1=vs1-s1;
    vector<int> v2;
    int ct=0;
    p=1;
    while(st.count(p)==0){
        st.insert(p);
        v2.push_back(p);
        if(ct==n)break;
        if(p-vs1>0){
            p=power(a,p-vs1,d1)+s1;
            p=v1[p];
        }
        else {
            p=power(a,p,m);
        }
        ct++;
    }
    //warr(v2);
    //write(p);
    ll s2=find(v2.begin(),v2.end(),p)-v2.begin();
    ll vs2=v2.size();
    ll d2=vs2-s2;
    if(ct!=n)write(v2[(n-vs2+d2)%d2+s2]);
    else write(p);
    return 0;
}
0