P9660 [ICPC 2021 Macao R] Pass the Ball!
题目描述
有 $n$ 个孩子和 $n$ 个球在玩游戏。孩子和球都从 $1$ 编号到 $n$。
游戏开始前,给出了 $n$ 个整数 $p_1, p_2, \cdots, p_n$。在游戏的每一轮中,孩子 $i$ 会把他手里的球传给孩子 $p_i$。保证没有孩子会把他手里的球传给自己,也就是说 $p_i \neq i$。此外,我们还知道在每一轮之后,每个孩子手里都会正好持有一个球。
设 $b_i$ 表示孩子 $i$ 所持有的球。在游戏开始时,孩子 $i$($1 \le i \le n$)将携带球 $i$,也就是说 $b_i=i$。你需要处理 $q$ 个查询。对于每个查询,你会得到一个整数 $k$,你需要计算在 $k$ 轮后 $\sum\limits_{i=1}^{n} i \times b_i$ 的值。
输入格式
无
输出格式
无
说明/提示
The sample test case is explained below.
$$
\begin{array}{|c|c|c|c|c|c|} \hline \textbf{Round} & \textbf{b1} & \textbf{b2} & \textbf{b3} & \textbf{b4} & \textbf{Answer} \\\hline
1 & 3 & 1 & 4 & 2 & 25 \\\hline
2 & 4 & 3 & 2 & 1 & 20 \\\hline
3 & 2 & 4 & 1 & 3 & 25 \\\hline
4 & 1 & 2 & 3 & 4 & 30 \\\hline
\end{array}
$$