解题思路

由于序列 $a$ 是 $1 \sim n$ 的一个排列，每个节点的出度恰好为 $1$ ，入度也恰好为 $1$ ，因此整个有向图由若干个互不相交的环构成。
在同一个环中，从任意节点出发，沿着传送带移动，可以到达环内的所有其他节点（包括自身）。也就是说，环内任意有序对 $(p, q)$ 都是一个可达对。
设某个环包含 $x$ 个节点，则该环内部的可达对数目为 $x \cdot x = x^2$ 。
因此，可达对的总数 = 所有环的大小的平方之和。问题转化为：找出所有环的大小，计算它们的平方和。
具体实现：
- 用一个 visited 数组标记每个节点是否已经处理过。
- 遍历所有节点 $i = 1 \dots n$ ，若 visited[i] 为假，则从 $i$ 出发，一直沿 $a$ 向前走，直到回到节点 $i$ ，同时标记访问过的节点并计数。得到的计数即为当前环的大小 $c$ ，将 $c^2$ 累加到答案中。

题目内容

在一座庞大的迷宫中，有 $n$ 个平台，编号 $1$ 到 $n$ 。每个平台 $i$ 上设置了一条单向传送带，会将你运送到平台 $a_i$ 。已知序列 $a$ 包含了 $1$ 到 $n$ 的所有整数，且没有重复。

若从平台 $p$ 出发，通过若干次传送带移动（可以为零次）可以抵达平台 $q$ ，则称 $(p,q)$ 为一个可达对。请你计算可达对的总数目。

数据范围：平台数 $n$ 满足 $n \le 10^5$ ，序列 $a$ 中的整数均在 $1$ 到 $n$ 之间，且每个整数恰好出现一次。