C++の練習を兼ねて, AtCoder Beginner Contest 262 の 問題C (Min Max Pair) ～ 問題D (I Hate Non-integer Number) を解いてみた.
■感想.
1. 問題C, Dは, 方針を絞り込めたので, AC版に到達出来たと思う.
2. 問題Dで, 苦手な動的計画法の訓練を積めたので, 非常に良かったと思う.
3. 引き続き, 時間を見つけて, 過去問の学習を進めていきたいと思う.
本家のサイト AtCoder Beginner Contest 262 解説 の 各リンク を ご覧下さい.
■C++版プログラム(問題C／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 |
// 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--) int a[505050]; int main(){ // 1. 入力情報. int N; scanf("%d", &N); rep(i, N) scanf("%d", &a[i + 1]); // 2. カウント. LL s = 0, t = 0; repx(i, 1, N + 1){ if(i == a[i]) ++s; else if(i == a[a[i]]) ++t; } // 3. 出力. printf("%lld\n", s * (s - 1) / 2 + t / 2); 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 |
[入力例] 4 1 3 2 4 [出力例] 2 ※AtCoderテストケースより [入力例] 10 5 8 2 2 1 6 7 2 9 10 [出力例] 8 ※AtCoderテストケースより [入力例] 5 1 2 3 4 5 [出力例] 10 [入力例] 15 1 7 3 14 2 6 2 9 8 4 12 11 1 12 9 [出力例] 6 [入力例] 120 1 2 26 29 5 6 10 6 30 10 11 12 15 7 18 28 26 15 16 10 1 9 8 26 6 3 9 22 6 30 15 24 29 23 1 1 22 18 23 21 24 25 18 17 21 1 16 5 29 24 20 15 22 25 12 22 26 12 19 29 21 27 17 22 1 9 16 10 15 30 15 7 20 12 16 29 3 12 4 1 1 25 16 8 11 26 20 8 11 13 27 8 15 25 28 15 7 29 24 9 9 18 27 15 28 3 7 8 28 111 110 30 22 20 12 17 15 119 118 4 [出力例] 32 |
■C++版プログラム(問題D／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 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
// 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 = 998244353; int a[101]; int main(){ // 1. 入力情報. int N; scanf("%d", &N); rep(i, N) scanf("%d", &a[i]); // 2. dp更新. LL ans = 0; repx(p, 1, N + 1){ // i項目まで, j項を選ぶ場合に, 余りが k となる個数. LL dp[N + 1][N + 1][p]; rep(i, N + 1) rep(j, N + 1) rep(k, p) dp[i][j][k] = 0; dp[0][0][0] = dp[0][1][0] = 1; repx(i, 1, N + 1){ rep(j, i){ // a[i - 1] を 選ぶ. rep(k, p){ dp[i][j + 1][(k + a[i - 1]) % p] += dp[i - 1][j][k]; dp[i][j + 1][(k + a[i - 1]) % p] %= MOD; } // a[i - 1] を 選ばない. rep(k, p){ dp[i][j][k] += dp[i - 1][j][k]; dp[i][j][k] %= MOD; } } } // 集計. ans += dp[N][p][0]; ans %= MOD; } // 3. 出力. printf("%lld\n", ans); 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 38 39 40 41 42 43 |
[入力例] 3 2 6 2 [出力例] 6 ※AtCoderテストケースより [入力例] 5 5 5 5 5 5 [出力例] 31 ※AtCoderテストケースより [入力例] 7 1 2 3 4 5 6 7 [出力例] 45 [入力例] 15 11 2 3 9 13 9 10 12 2 13 12 8 14 4 12 [出力例] 4751 [入力例] 25 300 372 443 132 123 339 338 132 153 230 401 205 100 459 195 36 223 90 185 372 114 386 470 36 337 [出力例] 2809627 [入力例] 50 180 321 149 7 96 304 171 323 262 115 383 451 149 65 75 377 230 303 207 106 6 462 319 355 247 31 95 84 404 109 100 15 312 118 460 323 429 350 63 45 214 214 440 293 174 30 489 12 105 108 [出力例] 971158417 |