C++の練習を兼ねて, AtCoder Beginner Contest 168 の 問題D (.. (Double Dots)) を解いてみた.
■感想.
1. 前回訪問した頂点を, 保存出来たので, AC版になったと思う.
2. 幅優先探索の復習が出来たので, 非常に良かったと思う.
3. 時間を見つけて, 引き続き, 過去問を振り返っていきたいと思う.
本家のサイトABC 168 解説をご覧下さい.
■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 78 79 80 81 82 83 84 85 86 87 88 |
// C++(GCC 9.2.1) #include <bits/stdc++.h> using namespace std; using vvi = vector<vector<int>>; #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 #define a first #define b second const int LIMIT = 123456789; const int MAX = 202020; int d[MAX]; // 頂点 1 からの 距離 を 保存. int bef[MAX]; // 前回訪問した頂点. // グラフを幅優先探索する. // https://ja.wikipedia.org/wiki/幅優先探索 // ※bfsの動作確認用. // @param G: グラフ. // @param s: グラフの探索開始頂点. // @param d: 探索開始地点からの距離. // @return: 特に無し. void bfs(vvi &G, int s, int* d){ // 1. 空のキュー. queue<int> q; // 2. 訪問済みフラグ設定. d[s] = 0; // スタート地点は, 距離ゼロを指定. bef[s] = -1; // スタート地点は, 前回訪問した頂点が存在しないので, -1 を 指定. // 3. 探索地点 s をキュー q に追加. q.push(s); while(!q.empty()){ // 4. キューから取り出す. int u = q.front(); q.pop(); // 5. 取り出した要素を処理. for(auto &e : G[u]){ // 6. 訪問済であれば, 処理をスキップ. if(d[e] > 0) continue; if(d[e] == 0 && e != s) d[e] = d[u] + 1, q.push(e), bef[e] = u; } } return; } int main(){ // 1. 入力情報. int N, M, a, b; scanf("%d %d", &N, &M); vvi G(N); rep(i, M){ scanf("%d %d", &a, &b); a--, b--; G[a].pb(b); G[b].pb(a); } // 2. 初期化. rep(i, N) bef[i] = LIMIT; // 3. 頂点 1 から 各頂点までの距離を計算. bfs(G, 0, d); // rep(i, N) printf("%d ", d[i]); // puts(""); // 4. 訪問不可能な頂点があるかの確認. bool ok = true; rep(i, N){ if(bef[i] == LIMIT){ ok = false; break; } } // 5. 出力. if(ok){ puts("Yes"); repx(i, 1, N) printf("%d\n", bef[i] + 1); }else{ puts("No"); } 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 4 1 2 2 3 3 4 4 2 [出力例] Yes 1 2 2 ※AtCoderテストケースより [入力例] 6 9 3 4 6 1 2 4 5 3 4 6 1 5 6 2 4 5 5 6 [出力例] Yes 6 5 5 1 1 ※AtCoderテストケースより ※ 但し, 上記のプログラムでは, 以下の内容が出力される. Yes 6 5 6 1 1 [入力例] 5 4 1 2 3 4 4 5 3 5 [出力例] No [入力例] 5 6 1 2 1 4 2 4 3 2 4 3 5 4 [出力例] Yes 1 2 1 4 [入力例] 12 16 1 2 5 2 5 8 8 2 9 5 10 9 9 4 4 3 3 1 6 3 6 4 7 6 11 7 11 2 12 7 12 3 [出力例] Yes 1 1 3 2 3 11 2 5 9 2 3 |
■参照サイト
AtCoder Beginner Contest 168