C++の練習を兼ねて, AtCoder Beginner Contest 167 の 問題D (D – Teleporter) を解いてみた.
■感想.
1. 1回だけ訪問する町と, 2回以上訪問する町を分離出来たので, AC版に到達できたと思う.
2. 時間を見つけて, 引き続き, 過去問を振り返っていきたいと思う.
本家のサイトABC 167 解説(動画)をご覧下さい.
■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 |
#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--) #define pb push_back int a[202020], memo[202020]; int main(){ // 1. 入力情報. int N; LL K; scanf("%d %lld", &N, &K); rep(i, N) scanf("%d", &a[i + 1]); // 2. 周期を求める. int cur = 1; // 町1 が スタート. vector<int> v1, v2; // 訪問した町を記録. while(memo[cur] < 2){ // 訪問したことを記録. memo[cur]++; // テレポートした町を保存. if(memo[cur] == 1) v1.pb(cur); // 1回目訪問. if(memo[cur] == 2) v2.pb(cur); // 2回目訪問. // 次の町へ. cur = a[cur]; } // 3. 1回訪問した町 と 2回訪問した町を分ける. LL one = (LL)(v1.size() - v2.size()); LL two = (LL)v2.size(); // 4. K回テレポートした場合は？ int ans = 0; if(one >= K){ ans = v1[(int)K]; }else{ LL res = (K - one) % two; ans = v2[(int)res]; } // 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 |
[入力例] 4 5 3 2 4 1 [出力例] 4 ※AtCoderテストケースより [入力例] 6 727202214173249351 6 5 2 5 3 2 [出力例] 2 ※AtCoderテストケースより [入力例] 7 6 3 7 5 2 6 4 1 [出力例] 7 [入力例] 10 314159265358979323 8 9 2 6 10 4 3 7 5 2 [出力例] 10 [入力例] 100 123456789987654321 33 50 67 56 84 89 96 83 95 68 8 86 36 89 27 29 88 88 25 78 3 74 33 46 84 51 68 81 53 83 1 44 48 9 94 8 64 28 37 4 44 25 96 95 82 91 10 40 45 17 56 27 43 6 84 4 87 79 94 15 83 52 95 50 19 91 64 47 27 97 40 58 30 14 37 11 94 64 75 54 49 96 2 98 4 10 16 56 71 88 41 88 1 70 75 23 77 51 35 98 [出力例] 56 |
■参照サイト
AtCoder Beginner Contest 167