HackerRank 20/20 Hack November 3問目 - Power of large numbers

Source

HackerRank 20/20 Hack November 3問目
problem statement（コンテストページ）
problem statement

問題概要

$A^B\ {\rm mod}\ 1000000007\ (10^9+7)$ を求める問題．

解法

$10^9+7$ は素数だから，（$k \neq 0\ ({\rm mod}\ 1000000007)$ であれば）$k^{1000000006}\ {\rm mod}\ (10^9+7)$ は $1$ になる．
そこで，$A$ は ${\rm mod}\ 1000000007$ しておいて，$B$ は ${\rm mod}\ 1000000006$ しておいてから，べき乗を繰り返し自乗法（バイナリ法）などで求めれば良い．
Editorialの解法が $A=10^9+7, B=10^9+6$ の時などに間違えることを指摘したら，$A\ {\rm mod}\ (10^9+7) \neq 0$ という制約が追加された．

C++によるスパゲッティなソースコード

#include<bits/stdc++.h>
using namespace std;

#define REP(i,a,b) for(i=a;i<b;i++)
#define rep(i,n) REP(i,0,n)

#define ll long long
#define ull unsigned ll
#define M 1000000007

#define mygc(c) (c)=getchar_unlocked()
#define mypc(c) putchar_unlocked(c)

void reader(int *x){int k,m=0;*x=0;for(;;){mygc(k);if(k=='-'){m=1;break;}if('0'<=k&&k<='9'){*x=k-'0';break;}}for(;;){mygc(k);if(k<'0'||k>'9')break;*x=(*x)*10+k-'0';}if(m)(*x)=-(*x);}
int reader(char c[]){int i,s=0;for(;;){mygc(i);if(i!=' '&&i!='\n'&&i!='\r'&&i!='\t'&&i!=EOF) break;}c[s++]=i;for(;;){mygc(i);if(i==' '||i=='\n'||i=='\r'||i=='\t'||i==EOF) break;c[s++]=i;}return s;}
void writer(int x, char c){int i,sz=0,m=0;char buf[10];if(x<0)m=1,x=-x;while(x)buf[sz++]=x%10,x/=10;if(!sz)buf[sz++]=0;if(m)mypc('-');while(sz--)mypc(buf[sz]+'0');mypc(c);}

ull pw(ull a, ull b, ull md){
  ull r = 1;
  while(b){
    if(b&1) r = r*a%md;
    b>>=1;
    a = a*a%md;
  }
  return r;
}

int main(){
  int i, T;
  ll A, B;
  int res;
  char buf[100010]; int len;

  reader(&T);
  while(T--){
    A = B = 0;
    len = reader(buf);
    rep(i,len) A = (A*10+buf[i]-'0')%M;
    len = reader(buf);
    rep(i,len) B = (B*10+buf[i]-'0')%(M-1);

    if(A==0) res = 0; else res = pw(A,B,M);
    writer(res, '\n');
  }

  return 0;
}

Current time: 2024年04月23日19時09分38秒
Last modified: 2014年05月11日14時45分55秒 (by laycrs)
Tags: Competitive_Programming HackerRank HackerRank_nov13
トップページに戻る

Logged in as: unknown user (not login)

ログイン：