ATCoder [ABC167D] Teleporter

发布时间 2023-06-10 09:36:49作者: 语天QwQ

# 题目解析

这段代码的目标是处理一个含有 $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;
}