AT_abc175_d [ABC175D] Moving Piece

题目描述

高桥君打算在由编号为 $1, 2, \cdots, N$ 的 $N$ 个格子组成的棋盘上,用棋子玩一个游戏。每个格子 $i$ 上写有一个整数 $C_i$。此外,还给定了一个 $1, 2, \cdots, N$ 的排列 $P_1, P_2, \cdots, P_N$。 接下来,高桥君可以任选一个格子放置一个棋子,并且可以选择移动棋子的次数,次数不少于 $1$ 次且不超过 $K$ 次。每次移动的规则如下: - 每次移动时,若棋子当前在格子 $i$($1 \leq i \leq N$),则将棋子移动到格子 $P_i$。此时,分数会增加 $C_{P_i}$。 请你帮高桥君求出,游戏结束时可能获得的最大分数是多少。(游戏开始时分数为 $0$。)

输入格式

输入按以下格式从标准输入读入。 > $N$ $K$ > $P_1$ $P_2$ $\cdots$ $P_N$ > $C_1$ $C_2$ $\cdots$ $C_N$

输出格式

输出游戏结束时可能获得的最大分数。

说明/提示

## 限制条件 - $2 \leq N \leq 5000$ - $1 \leq K \leq 10^9$ - $1 \leq P_i \leq N$ - $P_i \neq i$ - $P_1, P_2, \cdots, P_N$ 互不相同 - $-10^9 \leq C_i \leq 10^9$ - 输入均为整数 ## 样例解释 1 任选一个格子开始,最多移动 $2$ 次的方案如下: - 初始在格子 $1$。移动 $1$ 次到格子 $2$,分数为 $4$。移动 $2$ 次到格子 $4$,分数为 $4 + (-8) = -4$。 - 初始在格子 $2$。移动 $1$ 次到格子 $4$,分数为 $-8$。移动 $2$ 次到格子 $1$,分数为 $-8 + 3 = -5$。 - 初始在格子 $3$。移动 $1$ 次到格子 $5$,分数为 $8$。移动 $2$ 次到格子 $3$,分数为 $8 + (-10) = -2$。 - 初始在格子 $4$。移动 $1$ 次到格子 $1$,分数为 $3$。移动 $2$ 次到格子 $2$,分数为 $3 + 4 = 7$。 - 初始在格子 $5$。移动 $1$ 次到格子 $3$,分数为 $-10$。移动 $2$ 次到格子 $5$,分数为 $-10 + 8 = -2$。 这些方案中的最大值为 $8$。 ## 样例解释 3 必须至少移动 $1$ 次棋子。 ## 样例解释 4 答案的绝对值可能会非常大。 由 ChatGPT 4.1 翻译