C++の練習を兼ねて, AtCoder Beginner Contest 211 の 問題C (chokudai) ~ 問題D (Number of Shortest paths) を解いてみた.
■感想.
1. 問題D は, 実装してみたもののAC版到達できなかったので, 解説を参考にして, ようやく, AC版に到達出来た.
2. 問題C は, 過去問 008 – AtCounter に似ていると感じた.
3. 幅優先探索(応用版, 経路数カウント)の復習が出来たので, 非常に良かったと思う.
4. 引き続き, 時間を見つけて, 過去問の学習を進めていきたいと思う.
本家のサイト AtCoder Beginner Contest 211 解説 の 各リンク を ご覧下さい.
■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 30 31 32 33 34 35 36 37 38 39 40 41 |
// 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 dp[101010][9]; int main(){ // 1. 入力情報. char c[101010]; scanf("%s", c); string S(c); int l = S.size(); // 2. dp更新. dp[0][0] = 1; rep(i, l){ rep(j, 9){ dp[i + 1][j] += dp[i][j]; if(S[i] == 'c' && j == 0) dp[i + 1][j + 1] += dp[i][j]; if(S[i] == 'h' && j == 1) dp[i + 1][j + 1] += dp[i][j]; if(S[i] == 'o' && j == 2) dp[i + 1][j + 1] += dp[i][j]; if(S[i] == 'k' && j == 3) dp[i + 1][j + 1] += dp[i][j]; if(S[i] == 'u' && j == 4) dp[i + 1][j + 1] += dp[i][j]; if(S[i] == 'd' && j == 5) dp[i + 1][j + 1] += dp[i][j]; if(S[i] == 'a' && j == 6) dp[i + 1][j + 1] += dp[i][j]; if(S[i] == 'i' && j == 7) dp[i + 1][j + 1] += dp[i][j]; } rep(j, 9) dp[i + 1][j] %= MOD; } // 3. 出力. printf("%lld\n", dp[l][8]); 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 |
[入力例] chchokudai [出力例] 3 ※AtCoderテストケースより [入力例] atcoderrr [出力例] 0 ※AtCoderテストケースより [入力例] chokudaichokudaichokudai [出力例] 45 ※AtCoderテストケースより [入力例] uciaoyatofazncmepcaptpoewbuhwqifkrssjbdfgrqoboelqpkvnchcxuxvpmzgcbydwfzvgxwjliqfzxdcqzggssmoeimllvhexhmqddjqrwabrmwteklwkghmocgescxinyflxsgauoivrmhjsrutmxurdchrvrpilxyaoadodirwgshoifajbacqechyyjwrqobj [出力例] 720 [入力例] tqrsvhwwyftvwfuuaswulzjypjxdrvjcjqgjpkeowqqrelmsxohutlovxvvkfrvduzxryffnxrufdoruiilpzlqrblgydotbgtjuwuftdaknaadmvwnovabhtbcxyggjsjpjpnuoxgqgafpkjhxorlmcbedmcbdjvpbwdpygeuqzfdcosezhvlanunivnmhdyccxanlghtxaptrmpusjrcqrwoolysekniigqdidndhgqkmzvnmudbfignecqfputmskyotcsfufxdykdphmsuwajyjexnhmdgzypxpkdtrjqaaeguxmmwymaryqtadzbcjeassymyjqcgnpgreseanutdgytwwrjgreswzaeetoymribmwijiqxjgzugzyfgqcbrkceutzmixinerlhgkataypocxkbmxolewvvxfeyhehrzwzifqyuxbqnsvbpwndasarjbjdknefazrxvdcojfmzejesyrzthndnlryyhdoujtijq [出力例] 227938 |
■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 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
// 解き直し. // https://atcoder.jp/contests/abc211/editorial/2003. // C++(GCC 9.2.1) #include <bits/stdc++.h> using namespace std; using LL = long long; using vi = vector<int>; using vvi = vector<vi>; #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--) #define pb push_back const LL MOD = 1e9 + 7; LL ans[202020]; int main(){ // 1. 入力情報. int N, M; scanf("%d %d", &N, &M); vvi G(N); rep(i, M){ int a, b; scanf("%d %d", &a, &b); a--, b--; G[a].pb(b); G[b].pb(a); } // 2. 都市 1 を スタート地点として探索. // https://ja.wikipedia.org/wiki/幅優先探索 auto bfs = [&](vvi &G, int s, int *d){ // 2-1. 空のキュー. queue<int> q; // 2-2. 探索地点 s をキュー q に追加. q.push(s); while(!q.empty()){ // 2-3. キューから取り出す. int u = q.front(); q.pop(); // 2-4. 取り出した要素を処理(※解説通り). for(auto &e : G[u]){ // 到達済み. if(d[e]){ if(d[e] == d[u] + 1){ ans[e] += ans[u]; ans[e] %= MOD; } } // 未到達. if(!d[e] && e != s){ d[e] = d[u] + 1; q.push(e); ans[e] = ans[u]; } } } }; int d[202020]; rep(i, 202020) d[i] = 0; ans[0] = 1; bfs(G, 0, d); // 3. 移動できない場合. if(!d[N - 1]){ puts("0"); return 0; } // 4. 出力. printf("%lld\n", ans[N - 1]); 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 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 |
[入力例] 4 5 2 4 1 2 2 3 1 3 3 4 [出力例] 2 ※AtCoderテストケースより [入力例] 4 3 1 3 2 3 2 4 [出力例] 1 ※AtCoderテストケースより [入力例] 2 0 [出力例] 0 ※AtCoderテストケースより [入力例] 7 8 1 3 1 4 2 3 2 4 2 5 2 6 5 7 6 7 [出力例] 4 ※AtCoderテストケースより [入力例] 15 20 1 3 1 2 2 3 2 4 2 5 4 9 4 10 5 7 5 6 7 10 10 11 7 11 6 8 8 11 11 12 8 12 8 13 13 15 12 14 14 15 [出力例] 1 [入力例] 7 7 1 3 1 4 2 3 2 4 2 5 5 6 2 7 [出力例] 2 [入力例] 10 13 1 5 2 5 1 6 2 6 2 4 3 6 3 4 3 7 4 8 7 8 7 9 8 10 9 10 [出力例] 5 |
■参照サイト
AtCoder Beginner Contest 211