C++の練習を兼ねて, AtCoder Regular Contest 122 の 問題A (Many Formulae) を解いてみた.
■感想.
1. 規則性を抽出できたので, AC版に到達出来たと思う.
2. 苦手な動的計画法の訓練を積めたので, 非常に良かったと思う.
3. 引き続き, 時間を見つけて, 過去問の学習を進めていきたいと思う.
本家のサイト AtCoder Regular Contest 122 解説 の 各リンク を ご覧下さい.
■C++版プログラム(問題A/AC版).
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
// C++(GCC 9.2.1) #include <bits/stdc++.h> using namespace std; using LL = long long; #define repex(i, a, b, c) for(int i = a; i < b; i += c) #define repx(i, a, b) repex(i, a, b, 1) #define rep(i, n) repx(i, 0, n) #define repr(i, a, b) for(int i = a; i >= b; i--) const LL MOD = 1e9 + 7; LL a[101010], f[101010], dp[101010][2]; int main(){ // 1. 入力情報. int N; scanf("%d", &N); rep(i, N) scanf("%lld", &a[i]); // 2. フィボナッチ数列. f[0] = 1; f[1] = 1; rep(i, N) f[i + 2] = f[i + 1] + f[i], f[i + 2] %= MOD; // 3. dp更新. dp[0][0] = a[0]; repx(i, 1, N){ dp[i][0] = (dp[i - 1][0] + dp[i - 1][1]) + a[i] * f[i]; dp[i][1] = dp[i - 1][0] + MOD * f[i - 1] - a[i] * f[i - 1]; dp[i][0] %= MOD; dp[i][1] %= MOD; } // 4. 出力. printf("%lld\n", (dp[N - 1][0] + dp[N - 1][1]) % MOD); return 0; } |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
[入力例] 3 3 1 5 [出力例] 15 ※AtCoderのテストケースより [入力例] 4 1 1 1 1 [出力例] 10 ※AtCoderのテストケースより [入力例] 10 866111664 178537096 844917655 218662351 383133839 231371336 353498483 865935868 472381277 579910117 [出力例] 279919144 ※AtCoderのテストケースより [入力例] 6 3 1 4 1 5 9 [出力例] 137 [入力例] 15 33 111 86 66 150 116 125 21 88 32 59 87 80 62 13 [出力例] 500033 |