結果

問題 No.978 Fibonacci Convolution Easy
ユーザー Ngô Lê Hoàng
提出日時 2024-01-05 19:00:56
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,708 bytes
コンパイル時間 1,574 ms
コンパイル使用メモリ 168,820 KB
実行使用メモリ 10,840 KB
最終ジャッジ日時 2024-09-27 19:09:20
合計ジャッジ時間 3,749 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 2 WA * 13 RE * 6
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define all(x) x.begin(),x.end()
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define rep_r(i,a,b) for(int i=a;i>=b;i--)
#define each(a,x) for (auto& x : a)
using pi = pair<int,int>;
using pl = pair<ll,ll>;
using vi = vector<int>;
using vl = vector<ll>;
#define fi first    
#define se second
#define sz(x) int(x.size())
#define so(x) sort(all(x))
#define so_r(x) sort(all(x),greater<int>())
#define zero(x) memset(x,0,sizeof(x))
#define pb push_back
#define lb lower_bound
#define ub upper_bound
const char nl = '\n';
int dx[4] = {1,-1,0,0};
int dy[4] = {0,0,1,-1};
int bit_cnt(int x){
    return __builtin_popcount(x);
}
ll bex(ll a, ll b, ll mod = 1e9 + 7){ll res = 1LL; while(b){ if (b&1) res = res * a % mod; a = a * a % mod; b >>= 1;} return res;}
template<class t,class u> bool chmax(t&a,u b){if(a<b){a=b;return true;}else return false;}
template<class t,class u> bool chmin(t&a,u b){if(b<a){a=b;return true;}else return false;}

// armistcxy
const int MOD = 1e9 + 7; // 998244353
const int N = 1e6 + 5;
const ll INF = 1e18;
int a[N], pref[N];
int add(int& a, int b){
    int cur = (a + b) % MOD;
    if (cur < 0) cur += MOD;
    return cur;
}
void solve(){
    int n; cin >> n;
    int p; cin >> p;
    a[1] = 0, a[2] = 1;
    rep(i,3,n) a[i] = (p * a[i - 1] + a[i-2]) % MOD;
    rep(i,1,n) {
        pref[i] = (pref[i-1] + a[i]) % MOD;
    }
    ll ans = 0;
    rep(i,1,n){
        int cur = (a[i] * add(pref[n], -pref[i-1])) % MOD;
        ans = (ans + cur) % MOD;
    }
    cout << ans << nl;
}

int main(){
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);

    int t = 1; //cin >> t;
    while (t--){
        solve();
    }
}
0