# 题目解析
这段代码的目标是处理一个含有 $n$ 个元素的整数序列,根据一定的规则,重复操作 $k$ 次后,确定操作结束时位于序列哪个位置。
## 解题思路
1. **读取输入**:首先,我们读取输入的整数 $n$ 和 $k$ ,以及整数序列 `a`。我们需要对序列的每个元素减一,以适应从 0 开始的索引。
cin >> n >> k; vector<int> a(n); for(int i = 0; i < n; i++) { cin >> a[i]; a[i]--; }
2. **初始化变量**:我们使用一个变量 `t` 来记录当前位置,以及一个数组 `vis` 来记录每个位置第一次被访问的次序。
vector<int> vis(n, -1); int t = 0; vis[t] = 0;
3. **进行操作**:我们按照规则进行操作。如果在某次操作后,当前位置已经被访问过,则表示找到了循环。此时,我们可以计算出循环的周期 `p`,以及还需进行的操作次数 `r`,然后快速进行 `r` 次操作。如果在进行 $k$ 次操作后还没有找到循环,则直接输出当前位置。
for(int i = 1; i <= k; i++) { t = a[t]; if(vis[t] != -1) { int p = i - vis[t]; int r = (k - i) % p; for(int j = 0; j < r; j++) t = a[t]; break; } vis[t] = i; }
4. **输出结果**:完成所有操作后,输出最终的位置,注意输出时要加一,以适应从 1 开始的索引。
cout << t + 1;
总的来说,这段代码通过模拟操作的过程,并在发现循环时利用循环的性质进行优化,找出重复操作 $k$ 次后位于序列的哪个位置。
# 完整代码:
#include <bits/stdc++.h> using namespace std; int main() { ios::sync_with_stdio(false); cin.tie(0); long long n, k; cin >> n >> k; vector<int> a(n); for(int i = 0; i < n; i++) { cin >> a[i]; a[i]--; } vector<int> vis(n, -1); int t = 0; vis[t] = 0; for(int i = 1; i <= k; i++) { t = a[t]; if(vis[t] != -1) { int p = i - vis[t]; int r = (k - i) % p; for(int j = 0; j < r; j++) t = a[t]; break; } vis[t] = i; } cout << t + 1; return 0; }
标签:ATCoder,Teleporter,int,ABC167D,cin,vis,vector,序列,操作 From: https://www.cnblogs.com/YuTianQwQ/p/17470787.html