#690224 (C++17) No.891 隣接3項間の漸化式

提出ソース

結果

問題	No.891 隣接3項間の漸化式
ユーザー	MtSaka
提出日時	2021-08-15 18:55:54
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	2 ms / 2,000 ms
コード長	4,194 bytes
コンパイル時間	4,321 ms
コンパイル使用メモリ	266,508 KB
実行使用メモリ	6,820 KB
最終ジャッジ日時	2024-10-08 01:04:48
合計ジャッジ時間	5,687 ms
ジャッジサーバーID （参考情報）	judge4 / judge3

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,816 KB
testcase_01	AC	2 ms 6,820 KB
testcase_02	AC	2 ms 6,816 KB
testcase_03	AC	2 ms 6,816 KB
testcase_04	AC	1 ms 6,816 KB
testcase_05	AC	2 ms 6,816 KB
testcase_06	AC	1 ms 6,820 KB
testcase_07	AC	2 ms 6,816 KB
testcase_08	AC	1 ms 6,816 KB
testcase_09	AC	2 ms 6,816 KB
testcase_10	AC	2 ms 6,820 KB
testcase_11	AC	2 ms 6,816 KB
testcase_12	AC	2 ms 6,816 KB
testcase_13	AC	1 ms 6,816 KB
testcase_14	AC	1 ms 6,816 KB
testcase_15	AC	2 ms 6,820 KB
testcase_16	AC	2 ms 6,820 KB
testcase_17	AC	2 ms 6,816 KB
testcase_18	AC	2 ms 6,816 KB
testcase_19	AC	2 ms 6,816 KB
testcase_20	AC	1 ms 6,816 KB
testcase_21	AC	1 ms 6,816 KB
testcase_22	AC	1 ms 6,820 KB
testcase_23	AC	2 ms 6,820 KB
testcase_24	AC	2 ms 6,816 KB
testcase_25	AC	1 ms 6,820 KB
testcase_26	AC	2 ms 6,816 KB
testcase_27	AC	1 ms 6,816 KB
testcase_28	AC	2 ms 6,820 KB
testcase_29	AC	2 ms 6,820 KB
testcase_30	AC	2 ms 6,816 KB
testcase_31	AC	2 ms 6,816 KB
testcase_32	AC	2 ms 6,816 KB
testcase_33	AC	2 ms 6,816 KB
testcase_34	AC	2 ms 6,816 KB
testcase_35	AC	1 ms 6,820 KB
testcase_36	AC	2 ms 6,816 KB
testcase_37	AC	2 ms 6,816 KB
testcase_38	AC	2 ms 6,816 KB
testcase_39	AC	1 ms 6,816 KB
testcase_40	AC	1 ms 6,820 KB
testcase_41	AC	1 ms 6,820 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

//GIVE ME AC!!!!!!!!!!!!!!!!!
//#pragma GCC target("avx")
//#pragma GCC optimize("O3")
//#pragma GCC optimize("unroll-loops")
#include<bits/stdc++.h>
#define ll long long
#define ld long double
#define floatset() fixed<<setprecision(15)
#define all(n) n.begin(),n.end()
#define rall(n) n.rbegin(),n.rend()
#define rep(i, s, n) for (ll i = s; i < (ll)(n); i++)
#define pb push_back
#define eb emplace_back
#define max_(a) *max_element(all(a))
#define min_(a) *min_element(all(a))
#define INT(...) int __VA_ARGS__;scan(__VA_ARGS__)
#define LL(...) ll __VA_ARGS__;scan(__VA_ARGS__)
#define STR(...) string __VA_ARGS__;scan(__VA_ARGS__)
#define CHR(...) char __VA_ARGS__;scan(__VA_ARGS__)
#define DBL(...) double __VA_ARGS__;scan(__VA_ARGS__)
#define LD(...) ld __VA_ARGS__;scan(__VA_ARGS__)
using namespace std;
using vl=vector<ll>;
using vi=vector<int>;
using vs=vector<string>;
using vc=vector<char>;
using vvl=vector<vl>;
using P=pair<ll,ll>;
using vvc=vector<vc>;
using vd=vector<double>;
using vp=vector<P>;
using vb=vector<bool>;
const int dx[8]={1,0,-1,0,1,-1,-1,1};
const int dy[8]={0,1,0,-1,1,1,-1,-1};
const ll inf=2e18;
const ll MOD=1000000007;
const ll mod=998244353;
const double pi=acos(-1);
template<typename T1,typename T2 >
ostream &operator<<(ostream&os,const pair<T1,T2>&p) {
  os<<p.first<<" "<<p.second;
  return os;
}
template<typename T1,typename T2>
istream &operator>>(istream&is,pair<T1,T2>&p) {
  is>>p.first>>p.second;
  return is;
}
template<typename T>
ostream &operator<<(ostream&os,const vector<T>&v) {
  for(int i=0;i<(int)v.size();i++) {
    os<<v[i]<<(i+1!=v.size()?" ":"");
  }
  return os;
}
template<typename T>
istream &operator>>(istream&is,vector<T>&v) {
  for(T &in:v)is>>in;
  return is;
}
void scan(){}
template<class Head,class... Tail>
void scan(Head&head,Tail&... tail) {
  cin>>head;
  scan(tail...);
}
template<class T>
void print(const T &t) { cout << t << '\n'; }
template<class Head, class... Tail>
void print(const Head &head, const Tail &... tail) {
  cout << head << ' ';
  print(tail...);
}
template<class... T>
void fin(const T &... a) {
  print(a...);
  exit(0);
}
template<typename T>
ll sum_(vector<T>&v){
  ll res=0;
  for(auto &e:v)res+=e;
  return res;
}
template<typename T1,typename T2>
inline bool chmax(T1&a,T2 b){return a<b&&(a=b,true);}
template<typename T1,typename T2>
inline bool chmin(T1&a,T2 b){return a>b&&(a=b,true);}
#if __has_include(<atcoder/all>)
#include <atcoder/all>
using mint = atcoder::modint1000000007;
#endif
template< class T >
struct Matrix {
  vector<vector<T>>A;
  Matrix() {}
  Matrix(size_t n,size_t m):A(n,vector<T>(m,0)){}
  Matrix(size_t n):A(n,vector<T>(n,0)){};
  size_t height(){
    return (A.size());
  }
  size_t width(){
    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));
  }
  static Matrix E(size_t n){
    Matrix mat(n);
    for(int i=0;i<n;i++)mat[i][i]=1;
    return (mat);
  }
  Matrix &operator+=(Matrix&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[i][j];
    return (*this);
  }
  Matrix &operator-=(Matrix&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[i][j];
    return (*this);
  }
  Matrix &operator*=(Matrix&B) {
    size_t n=height(),m=B.width(),p=width();
    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]+=(*this)[i][k]*B[k][j];
    A.swap(C);
    return (*this);
  }
  Matrix &operator^=(long long k){
    Matrix B=Matrix::E(height());
    while(k>0){
      if(k&1)B*=*this;
      *this*=*this;
      k>>=1LL;
    }
    A.swap(B.A);
    return (*this);
  }
  Matrix operator+(Matrix&B){
    return (Matrix(*this)+=B);
  }
  Matrix operator-(Matrix &B){
    return (Matrix(*this)-=B);
  }
  Matrix operator*(Matrix&B){
    return (Matrix(*this)*=B);
  }
  Matrix operator^(long long k){
    return (Matrix(*this)^=k);
  }
};
int main(){
  LL(a,b,n);
  if(n==0)fin(0);
  Matrix<mint>x(2),y(2,1);
  y[0][0]=1,y[1][0]=0;
  x[0][0]=a,x[0][1]=b,x[1][0]=1;
  x^=(n-1);
  y*=x;
  fin(y[0][0].val());
}

yukicoder

結果

テストケース

ソースコード