C++の練習を兼ねて, AtCoder Regular Contest 056 の 問題A (みんなでワイワイみかん) ~ 問題B (駐車場) を解いてみた.
■感想.
1. 問題B は, 解答方針が, 全く見えなかったので, 解答を参照して, 実装して何とかAC版となった.
2. 実装に苦労した点は, どの情報について, コスト最小化をしていくかという点だった.
3. ダイクストラ法の復習が出来たので, 非常に良かったと思う.
4. 時間を見つけて, 引き続き, 過去問を振り返っていきたいと思う.
本家のサイトARC 056 解説をご覧下さい.
■C++版プログラム(問題A/AC版).
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
#include <bits/stdc++.h> using namespace std; using LL = long long; int main(){ // 1. 入力情報. LL A, B, K, L; scanf("%lld %lld %lld %lld", &A, &B, &K, &L); // 2. K個のみかんを購入するために必要な金額の最小値は? // 2-1. セット無し. LL ans = A * K; // 2-2. セット有り. LL q = K / L, r = K % L; ans = min({ans, q * B + r * A, (q + 1) * B}); // 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 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 |
[入力例] 3 7 10 3 [出力例] 24 ※AtCoderのテストケースより [入力例] 4 5 11 3 [出力例] 20 ※AtCoderのテストケースより [入力例] 3 8 3 3 [出力例] 8 ※AtCoderのテストケースより [入力例] 3 8 2 3 [出力例] 6 ※AtCoderのテストケースより [入力例] 5 35 7 5 [出力例] 35 [入力例] 5 35 8 5 [出力例] 40 [入力例] 5 20 7 5 [出力例] 30 [入力例] 5 20 8 5 [出力例] 35 [入力例] 5 20 9 5 [出力例] 40 [入力例] 5 20 10 5 [出力例] 40 |
■C++版プログラム(問題B/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 |
// 解き直し. // https://img.atcoder.jp/data/arc/056/editorial.pdf #include <bits/stdc++.h> using namespace std; using vvi = vector<vector<int>>; using P = pair<int, int>; using PQ = priority_queue<P>; #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 int cost[202020]; // S から ある頂点 i までのパスで, 頂点番号が最小のものの最大値. int memo[202020]; // 訪問済み頂点番号. // https://ja.wikipedia.org/wiki/ダイクストラ法 // グラフを最良優先探索する. // ※dijkstraの動作確認用. // @param G: グラフ. // @param s: 今回探索する頂点番号. // @return: 特に無し. void dijkstra(vvi &G, int s){ // 1. 初期化. PQ pq; // {頂点番号, コスト} // 2. 始点設定. pq.push({s, s}); cost[s] = s; memo[s] = 1; // printf("0. s=%d cost[s]=%d\n", s, cost[s]); // 3. キュー pq が 空になるまで, 探索. while(!pq.empty()){ // 4. 探索を継続するために, 頂点を取得. int u = pq.top().a; // 頂点番号. int c = pq.top().b; // コスト. pq.pop(); // 5. より小さいコストが設定済みなら, Skip. // printf("1. u=%d c=%d cost[u]=%d memo[u]=%d\n", u, c, cost[u], memo[u]); if(cost[u] < c) continue; // 6. 各隣接する頂点について, コストを確認し, 最短コストを更新. for(auto &v : G[u]){ // 最短コスト更新. int mCost = min(u, c); if(!memo[v] && cost[v] > mCost) cost[v] = mCost, pq.push({v, mCost}), memo[v] = 1; // printf("2. v=%d cost[v]=%d memo[v]=%d\n", v, cost[v], memo[v]); } } } int main(){ // 1. 入力情報. int N, M, S; scanf("%d %d %d", &N, &M, &S); S--; vvi G(N); rep(i, M){ int u, v; scanf("%d %d", &u, &v); u--, v--; G[u].pb(v); G[v].pb(u); } // 2. ダイクストラ法. rep(i, 202020) cost[i] = 2020202020; dijkstra(G, S); // rep(i, N) printf("%d ", cost[i]); // puts(""); // 3. 出力. rep(i, N) if(cost[i] >= i) printf("%d\n", 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 139 140 141 142 143 |
[入力例] 3 3 2 1 2 2 3 1 3 [出力例] 1 2 ※AtCoderのテストケースより [入力例] 5 6 5 1 5 3 5 3 2 4 1 1 2 4 3 [出力例] 1 2 3 5 ※AtCoderのテストケースより [入力例] 5 5 5 1 4 4 3 3 2 2 5 5 1 [出力例] 1 2 5 ※AtCoderのテストケースより [入力例] 7 8 6 6 5 6 3 5 2 1 5 3 2 3 1 1 4 2 7 [出力例] 1 2 3 5 6 [入力例] 2 1 2 1 2 [出力例] 1 2 [入力例] 2 1 1 1 2 [出力例] 1 [入力例] 6 7 3 3 1 3 4 1 5 4 5 4 2 2 6 6 5 [出力例] 1 2 3 [入力例] 11 11 10 10 1 1 4 1 5 10 7 10 8 8 11 11 9 7 5 7 3 3 6 3 2 [出力例] 1 2 3 5 7 8 10 [入力例] 16 17 15 15 16 15 11 15 12 12 10 12 8 8 7 12 2 2 6 12 14 11 14 2 1 1 4 5 1 11 13 11 5 5 3 5 9 [出力例] 1 2 3 5 7 8 10 11 12 15 |
■参照サイト
AtCoder Regular Contest 056