C++の練習を兼ねて, AtCoder Beginner Contest 273 の 問題C ((K+1)-th Largest Number) ~ 問題D (LRUD Instructions) を解いてみた.
■感想.
1. 問題C, Dは, 方針を絞り込めたので, AC版に到達できたと思う.
2. 引き続き, 時間を見つけて, 過去問の学習を進めていきたいと思う.
本家のサイト AtCoder Beginner Contest 273 解説 の 各リンク を ご覧下さい.
■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 |
// C++(GCC 9.2.1) #include <bits/stdc++.h> using namespace std; #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 a first #define b second int a[202020], b[202020], ans[202020]; int main(){ // 1. 入力情報. int N; scanf("%d", &N); map<int, int> o; rep(i, N){ scanf("%d", &a[i]); o[a[i]] = 0; } // 2. 座標圧縮. int idx = 0; for(auto &p : o) p.b = ++idx; // 3. 種類数. rep(i, N) b[i] = idx - o[a[i]]; rep(i, N) ++ans[b[i]]; // 4. 出力. rep(i, N) printf("%d\n", ans[i]); 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 |
[入力例] 6 2 7 1 8 2 8 [出力例] 2 1 2 1 0 0 ※AtCoderテストケースより [入力例] 1 1 [出力例] 1 ※AtCoderテストケースより [入力例] 10 979861204 57882493 979861204 447672230 644706927 710511029 763027379 710511029 447672230 136397527 [出力例] 2 1 2 1 2 1 1 0 0 0 ※AtCoderテストケースより [入力例] 10 1234 1233 1234 1235 1234 1233 1235 1235 1236 1232 [出力例] 1 3 3 2 1 0 0 0 0 0 [入力例] 30 12 13 7 2 3 14 6 10 9 13 3 2 12 1 9 3 11 15 13 9 3 6 2 1 14 12 9 1 8 1 [出力例] 1 2 3 3 1 1 4 1 1 2 4 3 4 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 |
■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 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 |
// C++(GCC 9.2.1) #include <bits/stdc++.h> using namespace std; using vi = vector<int>; using vvi = vector<vi>; using P = pair<int, int>; using vp = vector<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 #define all(x) x.begin(), x.end() int main(){ // 1. 入力情報. int H, W, rs, cs, N; scanf("%d %d %d %d %d", &H, &W, &rs, &cs, &N); vp v(N); map<int, int> mr, mc; rep(i, N){ scanf("%d %d", &v[i].a, &v[i].b); mr[v[i].a] = 0; mc[v[i].b] = 0; } // 2. 座標圧縮. // 2-1. 行方向. int rIdx = -1; for(auto &p : mr) p.b = ++rIdx; // 2-2. 列方向. int cIdx = -1; for(auto &p : mc) p.b = ++cIdx; // 3. 壁情報. vvi gr(rIdx + 1), gc(cIdx + 1); rep(i, N){ gr[mr[v[i].a]].pb(v[i].b); gc[mc[v[i].b]].pb(v[i].a); } // 4. sort. // 4-1. 行方向. rep(i, rIdx + 1){ gr[i].pb(0); gr[i].pb(W + 1); sort(all(gr[i])); } // 4-2. 列方向. rep(i, cIdx + 1){ gc[i].pb(0); gc[i].pb(H + 1); sort(all(gc[i])); } // 5. クエリ回答. int Q; scanf("%d", &Q); P cur = {rs, cs}; rep(i, Q){ // クエリ入力情報. char d[11]; int l; scanf("%s %d", d, &l); // 上. if(d[0] == 'U'){ if(mc.count(cur.b)){ int rAt = lower_bound(all(gc[mc[cur.b]]), cur.a) - gc[mc[cur.b]].begin(); --rAt; cur.a = max(gc[mc[cur.b]][rAt] + 1, cur.a - l); }else{ cur.a = max(1, cur.a - l); } } // 下. if(d[0] == 'D'){ if(mc.count(cur.b)){ int rAt = lower_bound(all(gc[mc[cur.b]]), cur.a) - gc[mc[cur.b]].begin(); cur.a = min(gc[mc[cur.b]][rAt] - 1, cur.a + l); }else{ cur.a = min(H, cur.a + l); } } // 左. if(d[0] == 'L'){ if(mr.count(cur.a)){ int cAt = lower_bound(all(gr[mr[cur.a]]), cur.b) - gr[mr[cur.a]].begin(); --cAt; cur.b = max(gr[mr[cur.a]][cAt] + 1, cur.b - l); }else{ cur.b = max(1, cur.b - l); } } // 右. if(d[0] == 'R'){ if(mr.count(cur.a)){ int cAt = lower_bound(all(gr[mr[cur.a]]), cur.b) - gr[mr[cur.a]].begin(); cur.b = min(gr[mr[cur.a]][cAt] - 1, cur.b + l); }else{ cur.b = min(W, cur.b + l); } } // 出力. printf("%d %d\n", cur.a, cur.b); } 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 |
[入力例] 5 5 4 4 3 5 3 2 2 1 4 4 L 2 U 3 L 2 R 4 [出力例] 4 2 3 2 3 1 3 5 ※AtCoderテストケースより [入力例] 6 6 6 3 7 3 1 4 3 2 6 3 4 5 5 1 1 3 2 10 D 3 U 3 L 2 D 2 U 3 D 3 U 3 R 3 L 3 D 1 [出力例] 6 3 5 3 5 1 6 1 4 1 6 1 4 1 4 2 4 1 5 1 ※AtCoderテストケースより [入力例] 15 20 5 5 20 1 10 2 6 2 20 3 3 3 17 4 10 7 2 7 8 7 11 7 13 10 7 10 19 11 13 12 1 12 3 12 7 13 10 14 20 15 5 15 16 18 D 2 R 3 D 3 L 4 D 4 R 2 D 5 R 8 U 5 R 8 D 2 L 10 U 7 L 1 U 5 L 5 U 2 R 12 [出力例] 7 5 7 7 9 7 9 3 11 3 11 5 14 5 14 13 12 13 12 20 13 20 13 11 8 11 8 10 5 10 5 5 3 5 3 16 |
■参照サイト
パナソニックグループプログラミングコンテスト2022(AtCoder Beginner Contest 273)