結果

問題 No.978 Fibonacci Convolution Easy
ユーザー set0gut1set0gut1
提出日時 2020-01-31 21:40:17
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 46 ms / 2,000 ms
コード長 830 bytes
コンパイル時間 814 ms
コンパイル使用メモリ 86,312 KB
実行使用メモリ 34,532 KB
最終ジャッジ日時 2024-09-18 20:52:26
合計ジャッジ時間 1,886 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <algorithm>
#include <bitset>
#include <iostream>
#include <map>
#include <numeric>
#include <queue>
#include <set>
#include <string>
#include <unordered_map>
#include <unordered_set>
#include <vector>
#define MOD (1000000007l)
#define ll long long
#define rep(i, n) for (ll i = 0; i < (n); i++)

using namespace std;

void solve() {
  ll N, P;
  cin >> N >> P;
  vector<ll> A(N);
  A[0] = 0;
  A[1] = 1;
  for (ll i = 2; i < N; i++) {
    ll tmp = P * A[i-1] + A[i-2];
    tmp %= MOD;
    A[i] = tmp;
  }
  vector<ll> B(N);
  for (ll i = 1; i < N; i++) {
    B[i] = (B[i-1] + A[i]) % MOD;
  }

  ll ans = 0;
  rep (i, N) {
    ans += B[i] * A[i];
    ans %= MOD;
  }

  cout << ans << endl;
}

int main(void) {
  cin.tie(0);
  ios::sync_with_stdio(false);
  cout.precision(12);
  cout << fixed;
  solve();
  return 0;
}
0