C++の練習を兼ねて, AtCoder Beginner Contest 303 の 問題E (A Gift From the Stars) を解いてみた.
■感想.
1. 問題Eは, 方針を絞り込めたので, AC版に到達できたと思う.
2. 個人的には, グラフの性質を復習できたので, 非常に良かったと思う.
3. 引き続き, 時間を見つけて, 過去問の学習を進めていきたいと思う.
本家のサイト AtCoder Beginner Contest 303 解説 の 各リンク を ご覧下さい.
■C++版プログラム(問題E／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 |
// C++(GCC 9.2.1) #include <bits/stdc++.h> using namespace std; using vi = 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 all(x) x.begin(), x.end() #define pb push_back int main(){ // 1. 入力情報. int N; scanf("%d", &N); set<int> G[N]; rep(i, N - 1){ int u, v; scanf("%d %d", &u, &v); --u, --v; G[u].insert(v); G[v].insert(u); } // 2. 葉っぱを保存. queue<int> q; rep(i, N) if(G[i].size() == 1) q.push(i); // 3. 星を抽出. vi ans; while(!q.empty()){ // キューから取り出す. int u = q.front(); q.pop(); // 次数がゼロか？ if(!G[u].size()) continue; // 星 の 中心. int s = *G[u].begin(); // レベル保存. ans.pb(G[s].size()); // 隣接頂点. for(auto &s1 : G[s]){ // 星を構成する辺を削除. G[s1].erase(s); // 星を構成する頂点を処理. for(auto &s2 : G[s1]){ // Skip. if(s == s2) continue; // 星どうしを結ぶ辺を削除. G[s2].erase(s1); // 新たな葉っぱを追加. if(G[s2].size() == 1) q.push(s2); } // 星を構成する頂点を削除. G[s1].clear(); } // 星 を 削除. G[s].clear(); } // 4. sort. sort(all(ans)); // 5. 出力. int n = ans.size(); rep(i, n) printf("%d%s", ans[i], (i < n - 1) ? " " : "\n"); 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 |
[入力例] 6 1 2 2 3 3 4 4 5 5 6 [出力例] 2 2 ※AtCoderテストケースより [入力例] 9 3 9 7 8 8 6 4 6 4 1 5 9 7 3 5 2 [出力例] 2 2 2 ※AtCoderテストケースより [入力例] 20 8 3 8 18 2 19 8 20 9 17 19 7 8 7 14 12 2 15 14 10 2 13 2 16 2 1 9 5 10 15 14 6 2 4 2 11 5 12 [出力例] 2 3 4 7 ※AtCoderテストケースより [入力例] 33 1 2 5 2 3 2 3 4 6 3 4 7 7 12 4 8 13 8 19 13 9 6 20 9 9 11 14 9 10 9 11 18 18 21 22 21 17 10 15 10 16 10 33 28 28 32 28 31 28 30 28 29 15 25 25 26 25 27 23 16 24 23 17 33 [出力例] 2 2 2 2 3 3 5 5 [入力例] 43 1 2 1 3 1 4 5 4 6 5 24 5 9 5 15 5 6 7 7 8 10 9 10 14 10 12 10 13 10 11 25 30 25 28 25 26 25 31 25 27 25 32 25 29 24 25 16 15 16 17 18 16 19 18 20 19 21 20 23 20 22 20 33 22 34 33 39 34 40 34 41 34 37 34 43 34 36 34 42 34 35 34 38 34 [出力例] 2 3 3 4 5 8 10 |
■参照サイト
日鉄ソリューションズプログラミングコンテスト2023（AtCoder Beginner Contest 303）