結果
問題 | No.978 Fibonacci Convolution Easy |
ユーザー |
![]() |
提出日時 | 2020-03-10 01:03:49 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 40 ms / 2,000 ms |
コード長 | 3,592 bytes |
コンパイル時間 | 1,789 ms |
コンパイル使用メモリ | 175,976 KB |
実行使用メモリ | 18,836 KB |
最終ジャッジ日時 | 2024-11-15 20:59:17 |
合計ジャッジ時間 | 2,900 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 21 |
ソースコード
#include<bits/stdc++.h> using namespace std; #define ALL(x) x.begin(),x.end() #define rep(i,n) for(int i=0;i<(n);i++) #define debug(v) cout<<#v<<":";for(auto x:v){cout<<x<<' ';}cout<<endl; #define INF 1000000000 #define mod 1000000007 using ll=long long; const ll LINF=1001002003004005006ll; int dx[]={1,0,-1,0}; int dy[]={0,1,0,-1}; ll gcd(ll a,ll b){return b?gcd(b,a%b):a;} template<class T>bool chmax(T &a,const T &b){if(a<b){a=b;return true;}return false;} template<class T>bool chmin(T &a,const T &b){if(b<a){a=b;return true;}return false;} template<class T> vector<vector<T>> matplus(vector<vector<T>> a,vector<vector<T>> b){ asert(a.size()==b.size() and a[0].size()==b[0].size()); for(int i=0;i<a.size();i++){ for(int j=0;j<a[0].size();j++){ a[i][j]+=b[i][j]; } } return a; } template<class T> vector<vector<T>> matminus(vector<vector<T>> a,vector<vector<T>> b){ asert(a.size()==b.size() and a[0].size()==b[0].size()); for(int i=0;i<a.size();i++){ for(int j=0;j<a[0].size();j++){ a[i][j]-=b[i][j]; } } return a; } template<class T> vector<vector<T>> matmul(vector<vector<T>> a,vector<vector<T>> b){ assert(a[0].size()==b.size()); int n=b.size(); vector<vector<T>> ret(a.size(),vector<T>(b[0].size(),0)); for(int i=0;i<a.size();i++){ for(int j=0;j<b[0].size();j++){ for(int k=0;k<n;k++){ ret[i][j]+=a[i][k]*b[k][j]; } } } return ret; } template<class T> vector<vector<T>> matpow(vector<vector<T>> a,ll k){ assert(a.size()==a[0].size()); int n=a.size(); vector<vector<T>> ret(n,vector<T>(n,0)); for(int i=0;i<n;i++) ret[i][i]=1; while(k>0){ if(k&1) ret=matmul(ret,a); a=matmul(a,a); k>>=1; } return ret; } //mod vector<vector<ll>> matplus_mod(vector<vector<ll>> a,vector<vector<ll>> b){ assert(a.size()==b.size() and a[0].size()==b[0].size()); for(int i=0;i<a.size();i++){ for(int j=0;j<a[0].size();j++){ a[i][j]+=b[i][j]; if(a[i][j]>=mod) a[i][j]-=mod; } } return a; } vector<vector<ll>> matminus_mod(vector<vector<ll>> a,vector<vector<ll>> b){ assert(a.size()==b.size() and a[0].size()==b[0].size()); for(int i=0;i<a.size();i++){ for(int j=0;j<a[0].size();j++){ a[i][j]-=b[i][j]; if(a[i][j]<0) a[i][j]+=mod; } } return a; } vector<vector<ll>> matmul_mod(vector<vector<ll>> a,vector<vector<ll>> b){ assert(a[0].size()==b.size()); int n=b.size(); vector<vector<ll>> ret(a.size(),vector<ll>(b[0].size(),0)); for(int i=0;i<a.size();i++){ for(int j=0;j<b[0].size();j++){ for(int k=0;k<n;k++){ ret[i][j]+=a[i][k]*b[k][j]%mod; ret[i][j]%=mod; } } } return ret; } vector<vector<ll>> matpow_mod(vector<vector<ll>> a,ll k){ assert(a.size()==a[0].size()); int n=a.size(); vector<vector<ll>> ret(n,vector<ll>(n,0)); for(int i=0;i<n;i++) ret[i][i]=1; while(k>0){ if(k&1) ret=matmul_mod(ret,a); a=matmul_mod(a,a); k>>=1; } return ret; } signed main(){ cin.tie(0); ios::sync_with_stdio(0); ll n,p;cin>>n>>p; vector<ll> a(n); a[0]=0;a[1]=1; for(int i=2;i<n;i++){ a[i]=a[i-1]*p%mod+a[i-2]; a[i]%=mod; } ll sum=0,ans=0; for(int i=n-1;i>=0;i--){ sum=(sum+a[i])%mod; ans=(ans+sum*a[i]%mod)%mod; } cout<<ans<<endl; return 0; }