結果

問題 No.978 Fibonacci Convolution Easy
ユーザー Ngô Lê HoàngNgô Lê Hoàng
提出日時 2024-01-05 19:00:56
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,708 bytes
コンパイル時間 1,826 ms
コンパイル使用メモリ 167,400 KB
実行使用メモリ 10,796 KB
最終ジャッジ日時 2024-01-05 19:01:01
合計ジャッジ時間 4,788 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 RE -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 RE -
testcase_08 WA -
testcase_09 RE -
testcase_10 RE -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 RE -
testcase_17 RE -
testcase_18 AC 2 ms
6,676 KB
testcase_19 AC 2 ms
6,676 KB
testcase_20 WA -
権限があれば一括ダウンロードができます

ソースコード

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