#529146 (C++17) No.1161 Many Powers

• 問題: No.1161 Many Powers
• ユーザー: beet
• 提出日時: 2020-08-13 16:50:21
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 126 ms / 2,000 ms
• コード長: 2,039 bytes
• コンパイル時間: 1,991 ms
• コンパイル使用メモリ: 193,980 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-04-18 01:38:11
• 合計ジャッジ時間: 3,599 ms
• ジャッジサーバーID （参考情報）: judge3 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	1 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	2 ms; 5,376 KB
testcase_05	AC	5 ms; 5,376 KB
testcase_06	AC	8 ms; 5,376 KB
testcase_07	AC	2 ms; 5,376 KB
testcase_08	AC	48 ms; 5,376 KB
testcase_09	AC	58 ms; 5,376 KB
testcase_10	AC	22 ms; 5,376 KB
testcase_11	AC	41 ms; 5,376 KB
testcase_12	AC	22 ms; 5,376 KB
testcase_13	AC	64 ms; 5,376 KB
testcase_14	AC	109 ms; 5,376 KB
testcase_15	AC	83 ms; 5,376 KB
testcase_16	AC	93 ms; 5,376 KB
testcase_17	AC	70 ms; 5,376 KB
testcase_18	AC	56 ms; 5,376 KB
testcase_19	AC	34 ms; 5,376 KB
testcase_20	AC	126 ms; 5,376 KB
testcase_21	AC	22 ms; 5,376 KB
testcase_22	AC	74 ms; 5,376 KB
testcase_23	AC	2 ms; 5,376 KB
testcase_24	AC	2 ms; 5,376 KB

ソースコード

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

using Int = long long;
const char newl = '\n';

template<typename T1,typename T2> inline void chmin(T1 &a,T2 b){if(a>b) a=b;}
template<typename T1,typename T2> inline void chmax(T1 &a,T2 b){if(a<b) a=b;}
template<typename T> void drop(const T &x){cout<<x<<endl;exit(0);}
template<typename T=Int>
vector<T> read(size_t n){
  vector<T> ts(n);
  for(size_t i=0;i<n;i++) cin>>ts[i];
  return ts;
}


template<typename T>
struct Rint{
  static T mod;
  static void set_mod(T nmod){mod=nmod;}

  T v;
  Rint():v(0){}
  Rint(signed v):v(v){}
  Rint(long long t){v=t%mod;if(v<0) v+=mod;}

  Rint pow(long long k){
    Rint res(1),tmp(v);
    while(k){
      if(k&1) res*=tmp;
      tmp*=tmp;
      k>>=1;
    }
    return res;
  }

  static Rint add_identity(){return Rint(0);}
  static Rint mul_identity(){return Rint(1);}

  Rint inv(){return pow(mod-2);}

  Rint& operator+=(Rint a){v+=a.v;if(v>=mod)v-=mod;return *this;}
  Rint& operator-=(Rint a){v+=mod-a.v;if(v>=mod)v-=mod;return *this;}
  Rint& operator*=(Rint a){v=1LL*v*a.v%mod;return *this;}
  Rint& operator/=(Rint a){return (*this)*=a.inv();}

  Rint operator+(Rint a) const{return Rint(v)+=a;}
  Rint operator-(Rint a) const{return Rint(v)-=a;}
  Rint operator*(Rint a) const{return Rint(v)*=a;}
  Rint operator/(Rint a) const{return Rint(v)/=a;}

  Rint operator-() const{return v?Rint(mod-v):Rint(v);}

  bool operator==(const Rint a)const{return v==a.v;}
  bool operator!=(const Rint a)const{return v!=a.v;}
  bool operator <(const Rint a)const{return v <a.v;}
};
template<typename T> T Rint<T>::mod;
template<typename T>
ostream& operator<<(ostream &os,Rint<T> m){os<<m.v;return os;}

//INSERT ABOVE HERE
signed main(){
  cin.tie(0);
  ios::sync_with_stdio(0);

  Int a,b,c;
  cin>>a>>b>>c;

  using R = Rint<Int>;
  R::set_mod(c);

  // [0, a] -> [0, a')
  a++;

  Int x=a/c;
  Int y=a%c;
  R ans{0};
  for(Int i=0;i<c;i++) ans+=R(i).pow(b);
  ans*=R(x);

  for(Int i=0;i<y;i++) ans+=R(i).pow(b);

  cout<<ans<<newl;
  return 0;
}