C++の練習を兼ねて, AtCoder Regular Contest 114 の 問題A (Not coprime) ~ 問題B (Special Subsets) を解いてみた.
■感想.
1. どちらも, 解答方針が見えなかったので, 解説を参考に実装して, ようやく, AC版に到達できた.
2. 幅優先探索の復習が出来たので, 非常に良かったと思う.
3. 時間を見つけて, 引き続き, 過去問を振り返っていきたいと思う.
本家のサイト AtCoder Regular Contest 114 解説 の 各リンク を ご覧下さい.
■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 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
// 解き直し. // https://atcoder.jp/contests/arc114/editorial/899 // C++(GCC 9.2.1) #include <bits/stdc++.h> using namespace std; using LL = long long; #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--) const LL p[] = {2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43, 47}; LL x[55]; int main(){ // 1. 入力情報. int N; scanf("%d", &N); rep(i, N) scanf("%lld", &x[i]); // 2. 全探索(解説通り). LL ans = 1e19; rep(i, 1 << 15){ LL Y = 1; rep(j, 15) if(i & (1 << j)) Y *= p[j]; bool ok = true; rep(j, N){ if(__gcd(x[j], Y) == 1){ ok = false; break; } } if(ok) ans = min(ans, Y); } // 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 |
[入力例] 2 4 3 [出力例] 6 ※AtCoderのテストケースより [入力例] 1 47 [出力例] 47 ※AtCoderのテストケースより [入力例] 7 3 4 6 7 8 9 10 [出力例] 42 ※AtCoderのテストケースより [入力例] 5 3 12 6 9 27 [出力例] 3 [入力例] 15 44 40 45 38 9 33 11 12 4 6 39 19 20 35 33 [出力例] 6270 [入力例] 30 48 4 38 38 44 24 26 39 50 43 16 27 48 28 30 6 20 34 12 27 47 37 14 35 32 10 44 23 22 46 [出力例] 51596130 [入力例] 45 13 3 34 6 9 47 48 37 30 2 49 23 16 11 43 15 47 48 5 25 32 20 22 24 8 29 17 6 3 42 13 47 18 19 13 39 9 23 29 8 41 33 32 36 31 [出力例] 614889782588491410 |
■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 |
// 解き直し. // https://atcoder.jp/contests/arc114/editorial/904 // C++(GCC 9.2.1) #include <bits/stdc++.h> using namespace std; using vi = vector<int>; using vvi = vector<vi>; using LL = long long; #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 LL MOD = 998244353; int main(){ // 1. 入力情報. int N; scanf("%d", &N); vvi G(N); rep(i, N){ int f; scanf("%d", &f); f--; G[i].pb(f); G[f].pb(i); } // 2. グラフを連結成分に分解. // https://ja.wikipedia.org/wiki/幅優先探索 auto bfs = [&](vvi &G, int s, int* c, int l){ // 2-1. 空のキュー. queue<int> q; // 2-2. 連結成分番号を設定. c[s] = l; // 2-3. 探索地点 s をキュー q に追加. q.push(s); while(!q.empty()){ // 2-4. キューから取り出す. int u = q.front(); q.pop(); // 2-5. 取り出した要素を処理. for(auto &e : G[u]){ // 2-6. 訪問済であれば, 処理をスキップ. if(c[e]) continue; if(!c[e] && e != s) c[e] = l, q.push(e); } } return; }; int c[N], idx = 0; memset(c, 0, sizeof(c)); rep(i, N) if(!c[i]) bfs(G, i, c, ++idx); // rep(i, N) printf("%d ", c[i]); // puts(""); // 3. 条件を満たす個数を計算. LL ans = 1; rep(i, idx) ans <<= 1, ans %= MOD; ans = (ans + MOD - 1) % MOD; // 4. 出力. 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 |
[入力例] 2 2 1 [出力例] 1 ※AtCoderのテストケースより [入力例] 2 1 1 [出力例] 1 ※AtCoderのテストケースより [入力例] 3 1 2 3 [出力例] 7 ※AtCoderのテストケースより [入力例] 17 2 1 4 5 3 7 8 9 6 11 10 13 14 15 16 12 17 [出力例] 63 [入力例] 50 1 3 2 1 7 8 5 9 6 5 12 11 13 15 14 3 19 17 18 22 21 20 24 25 26 27 23 11 31 29 30 18 33 35 34 40 36 37 38 39 8 43 44 42 50 45 49 47 46 48 [出力例] 131071 |
■参照サイト
AtCoder Regular Contest 114