C++の練習を兼ねて, AtCoder Beginner Contest 254 の 問題C (K Swap) ~ 問題D (Together Square) を解いてみた.
■感想.
1. 問題Dは, 方針が見えなかったので, 解説を参考に, AC版に到達できたと思う.
2. 引き続き, 時間を見つけて, 過去問の学習を進めていきたいと思う.
本家のサイト AtCoder Beginner Contest 254 解説 の 各リンク を ご覧下さい.
■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 36 37 38 39 |
// 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 #define all(x) x.begin(), x.end() int main(){ // 1. 入力情報. int N, K; scanf("%d %d", &N, &K); vi o; vvi t(K); rep(i, N){ int x; scanf("%d", &x); o.pb(x); t[i % K].pb(x); } // 2. 並び替え. sort(all(o)); rep(i, K) sort(all(t[i])); // 3. 復元. vi a(N); rep(i, K) rep(j, t[i].size()) a[i + j * K] = t[i][j]; // 4. 出力. printf("%s\n", (a == o) ? "Yes" : "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 |
[入力例] 5 2 3 4 1 3 4 [出力例] Yes ※AtCoderテストケースより [入力例] 5 3 3 4 1 3 4 [出力例] No ※AtCoderテストケースより [入力例] 7 5 1 2 3 4 5 5 10 [出力例] Yes ※AtCoderテストケースより [入力例] 3 1 3 2 1 [出力例] Yes [入力例] 3 1 2 1 3 [出力例] Yes [入力例] 7 2 6 7 1 3 2 4 9 [出力例] No [入力例] 7 2 10 7 6 5 3 2 1 [出力例] Yes |
■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 |
// 解き直し. // https://atcoder.jp/contests/abc254/editorial/4065 // 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--) int f[202020], g[202020]; int main(){ // 1. 入力情報. int N; scanf("%d", &N); // 2. f を 計算. int u = sqrt(N) + 1; repx(i, 1, u + 1){ // 基準とする平方数. int sq = i * i; // sq を約数に持つ整数で, f を 更新. repex(j, sq, N + 1, sq) f[j] = max(f[j], sq); } // 3. f[i] / i の カウントアップ. repx(i, 1, N + 1) ++g[i / f[i]]; // 4. 集計. int ans = N; rep(i, N + 1) ans += g[i] * (g[i] - 1); // 5. 出力. printf("%d\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 |
[入力例] 4 [出力例] 6 ※AtCoderテストケースより [入力例] 254 [出力例] 896 ※AtCoderテストケースより [入力例] 100 [出力例] 310 [入力例] 1000 [出力例] 4344 [入力例] 2022 [出力例] 9564 [入力例] 10000 [出力例] 57296 [入力例] 200000 [出力例] 1512912 |
■参照サイト
AtCoder Beginner Contest 254