#3094. Moving Piece

0
    ID: 3094 传统题 3000ms 1024MiB 尝试: 0 已通过: 0 难度: 4 上传者: 标签>AtCoderabc175DGESP 5级前缀和枚举(单层循环)

Moving Piece

Moving Piece

题目描述

Takahashi 将在编号为 1,2,⋯ ,N1, 2, \cdots, N 的格子数组上用一枚棋子进行游戏。第 ii 个格子上写着一个整数 CiC_i。此外,他得到一个 1,2,⋯ ,N1, 2, \cdots, N 的置换:P1,P2,⋯ ,PNP_1, P_2, \cdots, P_N。 现在,他将选择一个格子并把棋子放在该格子上。然后,他将进行以下操作若干次,次数在 11 到 KK(含)之间:

在一次移动中,如果棋子现在在第 ii 个格子上 (1≤i≤N)(1 \leq i \leq N),则将其移动到第 PiP_i 个格子。此时,他的得分增加 CPiC_{P_i}。

请你帮助他找到游戏结束时可能得到的最大得分。(游戏开始时得分为 00.)

输入格式

输入从标准输入按以下格式给出: NN KK P1P_1 P2P_2 ⋯\cdots PNP_N C1C_1 C2C_2 ⋯\cdots CNC_N

输出格式

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

5 2
2 4 5 1 3
3 4 -10 -8 8
8
2 3
2 1
10 -7
13
3 3
3 1 2
-1000 -2000 -3000
-1000
10 58
9 1 6 7 8 4 3 2 10 5
695279662 988782657 -119067776 382975538 -151885171 -177220596 -169777795 37619092 389386780 980092719
29507023469

提示

2≤N≤50002 \leq N \leq 5000 1≤K≤1091 \leq K \leq 10^9 1≤Pi≤N1 \leq P_i \leq N Pi≠iP_i \neq i P1,P2,⋯ ,PNP_1, P_2, \cdots, P_N 都互不相同。 −109≤Ci≤109-10^9 \leq C_i \leq 10^9 输入中的所有数值均为整数。当我们从任意选定的格子开始并最多进行两次移动时,我们有以下选项:

如果我们从格子 11 开始,进行一次移动会将棋子送到格子 22,此时得分为 44。再进行一次移动会将棋子送到格子 44,此时得分为 4+(−8)=−44 + (-8) = -4。 如果我们从格子 22 开始,进行一次移动会将棋子送到格子 44,此时得分为 −8-8。再进行一次移动会将棋子送到格子 11,此时得分为 −8+3=−5-8 + 3 = -5。 如果我们从格子 33 开始,进行一次移动会将棋子送到格子 55,此时得分为 88。再进行一次移动会将棋子送到格子 33,此时得分为 8+(−10)=−28 + (-10) = -2。 如果我们从格子 44 开始,进行一次移动会将棋子送到格子 11,此时得分为 33。再进行一次移动会将棋子送到格子 22,此时得分为 3+4=73 + 4 = 7。 如果我们从格子 55 开始,进行一次移动会将棋子送到格子 33,此时得分为 −10-10。再进行一次移动会将棋子送到格子 55,此时得分为 −10+8=−2-10 + 8 = -2。

能够达到的最大得分是 88.
我们必须至少进行一次移动。
答案的绝对值可能很大。

标签: AtCoder|abc175D

来源

AtCoder|abc175D