C++の練習を兼ねて, AtCoder Beginner Contest 142 の 問題F (Pure) を解いてみた.
■感想.
1. 問題F は, 方針が見えなかったので, 解説を参考に, AC版に到達できたと思う.
2. 幅優先探索, および, 経路復元 の 復習が出来たので, 非常に良かったと思う.
3. 引き続き, 時間を見つけて, 過去問の学習を進めていきたいと思う.
本家のサイト AtCoder Beginner Contest 142 解説 の 各リンク を ご覧下さい.
■C++版プログラム(問題F/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 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 |
// 解き直し. // https://atcoder.jp/contests/abc142/editorial/945 // C++(GCC 9.2.1) #include <bits/stdc++.h> using namespace std; 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 int INF = 1010101010; int main(){ // 1. 入力情報. int N, M; scanf("%d %d", &N, &M); vvi OG(N), RG(N); set<int> e[N]; rep(i, M){ int a, b; scanf("%d %d", &a, &b); --a, --b; OG[a].pb(b); // 有向グラフ. RG[b].pb(a); // 経路復元用のグラフ. e[b].insert(a); // 有向辺(b -> a). } // 2. bfs. // https://ja.wikipedia.org/wiki/幅優先探索 auto bfs = [&](vvi &G, int s, int* d){ // 空のキュー. queue<int> q; // 距離ゼロ. d[s] = 0; // 探索地点 s をキュー q に追加. q.push(s); while(!q.empty()){ // キューから取り出す. int u = q.front(); q.pop(); // 隣接頂点をチェック. for(auto &v : G[u]) if(d[v] == INF && v != s) d[v] = d[u] + 1, q.push(v); } }; // 3. 最短経路. auto f = [&](vvi &G, int s, int* d, vi& r){ // 空のキュー. queue<int> q; // 探索地点 s をキュー q に追加. q.push(s); r.pb(s); while(!q.empty()){ // キューから取り出す. int u = q.front(); q.pop(); // 取り出した要素を処理. for(auto &v : G[u]){ // 距離 が 1 小さい場合に, 保存. if(d[v] == d[u] - 1){ r.pb(v); q.push(v); break; } } } }; // 4. 全探索. vi ans; int dMax = INF; rep(i, N){ // 頂点 i を 始点として, bfs. int d[N]; rep(j, N) d[j] = INF; bfs(OG, i, d); // 頂点 i へ向かう辺に着目して, 最も距離が近い頂点 s を選択. int s = -1; for(auto &j : e[i]){ if(d[j] && d[j] < dMax){ dMax = d[j]; s = j; } } // 存在したら, 経路復元. if(s != -1){ vi cur; f(RG, s, d, cur); ans = cur; } } // 5. 出力. int K = ans.size(); if(!K){ puts("-1"); return 0; } printf("%d\n", K); repr(i, K - 1, 0) printf("%d\n", ans[i] + 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 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 |
[入力例] 4 5 1 2 2 3 2 4 4 1 4 3 [出力例] 3 1 2 4 ※AtCoderテストケースより [入力例] 4 5 1 2 2 3 2 4 1 4 4 3 [出力例] -1 ※AtCoderテストケースより [入力例] 6 9 1 2 2 3 3 4 4 5 5 6 5 1 5 2 6 1 6 2 [出力例] 4 2 3 4 5 ※AtCoderテストケースより [入力例] 10 10 1 2 2 8 8 1 2 3 2 4 10 3 3 7 9 7 4 5 6 5 [出力例] 3 1 2 8 [入力例] 13 15 9 1 1 2 2 4 4 6 2 5 6 7 5 7 7 8 8 3 3 10 8 10 10 2 11 7 12 11 13 11 [出力例] 5 2 5 7 8 10 [入力例] 32 34 1 3 3 4 1 2 2 5 5 6 2 7 25 7 7 8 8 9 9 10 11 10 12 13 13 14 10 14 15 14 14 16 16 19 19 17 18 19 19 20 21 20 20 22 22 23 23 24 24 25 31 32 32 23 10 28 28 29 29 30 30 25 10 26 26 27 27 25 [出力例] 7 7 8 9 10 26 27 25 |
■参照サイト
AtCoder Beginner Contest 142