AtCoder 4804 Beginner Contest 167 D Teleporter 思维题

最新推荐文章于 2020-09-18 00:37:20 发布

Curren.wong

最新推荐文章于 2020-09-18 00:37:20 发布

阅读量237

点赞数 1

分类专栏：算法刷题笔记 # 思维题文章标签：算法数据结构 c++

本文链接：https://blog.csdn.net/qq_41729780/article/details/106098757

版权

算法刷题笔记同时被 2 个专栏收录

76 篇文章 3 订阅

订阅专栏

思维题

3 篇文章 0 订阅

订阅专栏

文章目录

1. 题目描述

1.1. Limit

Time Limit: 2000 ms

Memory Limit: 1024 MB

1.2. Problem Description

The Kingdom of Takahashi has $N$ towns, numbered $1$ through $N$ .

There is one teleporter in each town. The teleporter in Town $i$ $\leq i \leq N)$ sends you to Town $A_i$ .

Takahashi, the king, loves the positive integer $K$ . The selfish king wonders what town he will be in if he starts at Town $1$ and uses a teleporter exactly $K$ times from there.

Help the king by writing a program that answers this question.

1.3. Input

$N$ $K$
$A_1\ A_2\ \dots \ A_N$

1.4. Constraints

$\leq N \leq 2 \times 10^5$

$\leq A_i \leq N$

$\leq K \leq 10^{18}$

1.5. Output

Print the integer representing the town the king will be in if he starts at Town $1$ and uses a teleporter exactly $K$ times from there.

1.6. Sample Input 1

4 5
3 2 4 1

1.7. Sample Output 1

If we start at Town $1$ and use the teleporter 5 times, our travel will be as follows: $\to 3 \to 4 \to 1 \to 3 \to 4$ .

1.8. Sample Input 2

6 727202214173249351
6 5 2 5 3 2

1.9. Sample Output 2

1.10. Source

AtCoder 4804 Teleporter

2. 解读

将所有路径使用邻接表进行存储，然后对路径依次进行遍历，将遍历过的节点存储在栈 $S t a c k$ 中，找出图中的环路。

在节点入栈前，判断其是否在栈中，若在，则该节点首次出现的位置到栈顶位置之间的节点所连成的边即为我们要找的环路 $C$ ，求出环路长度 $S$ 。

判断游走次数 $K$ 是否大于图的起点 $Q_0$ 到环路起点 $W_0$ 的距离 $length = Q_0 - W_0$ 。

若 $K > l e n g t h$ ，则输出 $S t a c k$ 中从 $0$ 开始计数第 $Q_0 + (K - W_0) \% S$ 个元素。

若 $\le length$ ，则输出 $S t a c k$ 中从 $0$ 开始计数第 $K$ 个元素。

3. 代码

#include <iostream>
#include <string.h>
#include <vector>
using namespace std;

const int NUM = 2 * 10e5 + 1;
// 存储
long long list[NUM];
// 标志访问
bool visit[NUM];
// 栈存储访问路径
vector<long long> vec;

int main()
{
    // test case
    long long n, k;
    scanf("%lld %lld", &n, &k);
    // 初始化
    memset(list, 0, sizeof(list));
    memset(visit, 0, sizeof(visit));
    // 输入
    for (long long i = 0; i < n; i++) {
        scanf("%lld", &list[i]);
    }
    // 初始化
    long long buffer = 1;
    // 计算
    for (int i = 0; i < n; i++) {
        // 标志访问
        visit[buffer] = 1;
        // 节点入栈
        vec.push_back(buffer);
        // 获取下一个节点
        buffer = list[buffer - 1];
        // 若形成环路
        if (visit[buffer] == 1) {
            break;
        }
    }
    // 在栈中找出环路开始位置
    auto it = find(vec.begin(), vec.end(), buffer);
    // 环路长度
    long long cycleLength = vec.end() - it;
    // 环路开始前的栈深度
    long long stackLength = it - vec.begin();
    // 计算从环路起点开始的位移
    long long posMark;
    if (k > stackLength) {
        // 若存在环路
        posMark = (k - stackLength) % cycleLength;
    } else {
        // 若不存在环路
        posMark = (k - stackLength);
    }
    // 输出
    printf("%lld\n", vec[stackLength + posMark]);
}