結果

問題 No.1706 Many Bus Stops (hard)
ユーザー TKTYI
提出日時 2021-10-08 23:25:17
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 3 ms / 2,000 ms
コード長 1,528 bytes
コンパイル時間 3,970 ms
コンパイル使用メモリ 236,220 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-07-23 07:26:05
合計ジャッジ時間 5,028 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
typedef long long int ll;
typedef long double ld;
#define FOR(i,l,r) for(ll i=l;i<r;i++)
#define REP(i,n) FOR(i,0,n)
#define RFOR(i,l,r) for(ll i=r-1;i>=l;i--)
#define RREP(i,n) RFOR(i,0,n)
#define ALL(x) x.begin(),x.end()
#define P pair<ll,ll>
#define F first
#define S second
#define BS(A,x) binary_search(ALL(A),x)
#define LB(A,x) (ll)(lower_bound(ALL(A),x)-A.begin())
#define UB(A,x) (ll)(upper_bound(ALL(A),x)-A.begin())
#define COU(A,x) UB(A,x)-LB(A,x)
template<typename T>using min_priority_queue=priority_queue<T,vector<T>,greater<T>>;
using mint=modint1000000007;
//using mint=modint998244353;
vector<vector<mint>>pow(vector<vector<mint>>A,ll N){
  vector<vector<mint>>res(4,vector<mint>(4)),p(4,vector<mint>(4));
  REP(i,4)res[i][i]=1;
  while(N){
    if(N%2){
      //res*=A
      REP(i,4)REP(j,4)p[i][j]=0;
      REP(i,4)REP(j,4)REP(k,4)p[i][j]+=res[i][k]*A[k][j];
      REP(i,4)REP(j,4)res[i][j]=p[i][j];
    }
    N/=2;
    //A*=A
    REP(i,4)REP(j,4)p[i][j]=0;
    REP(i,4)REP(j,4)REP(k,4)p[i][j]+=A[i][k]*A[k][j];
    REP(i,4)REP(j,4)A[i][j]=p[i][j];
  }
  return res;
}
signed main(){
  ll C,N,M;cin>>C>>N>>M;
  vector<vector<mint>>A(4,vector<mint>(4));
  A[0][0]=inv_mod(C,1000000007);
  A[0][1]=1;
  A[1][2]=A[0][0];
  A[2][2]=A[0][0];
  A[2][3]=1;
  A[3][0]=(C-1)*A[0][0];
  A[3][2]=(C-2)*A[0][0];
  mint ans=pow(A,N).at(0).at(0);
  ans=1-pow_mod((1-ans).val(),M,1000000007);
  cout<<ans.val()<<endl;
  return 0;
}
0