D. Captain Flint and Treasure(拓扑排序 + 贪心)

Problem - D - Codeforces

芬特队长参与了另一个寻宝活动，但只发现了一个奇怪的问题。这个问题可能与宝藏的位置有关，也可能不是。这就是为什么弗林特船长决定把解决问题的工作交给他的船员，并提供了一个高得离谱的奖励:休息一天。问题本身听起来是这样的……有两个长度为n的数组a和b。最初，ans等于0，并定义如下操作:1. 选择位置i (1 bb;?不是循环的，换句话说，它总是以-1结尾。输出在第一行中，打印您可以得到的最大ans。在第二行，打印操作顺序:n个不同的整数P1,P2,Pn (1 pi S n)。p是第i步应该选择的位置。如果有多个订单，打印其中任何一个。

Examples

input

Copy

3
1 2 3
2 3 -1

output

Copy

10
1 2 3 

input

Copy

2
-1 100
2 -1

output

Copy

99
2 1 

input

Copy

10
-10 -1 2 2 5 -2 -3 -4 2 -6
-1 -1 2 2 -1 5 5 7 7 9

output

Copy

-9
3 5 6 1 9 4 10 7 8 2

题解:

如果bi != -1,那么如果进行第i操作,abi += ai,bi可以看作一条i -> bi的边,

又说保证是不循环的,所以构建出来的图大概是这样

由于 abi += ai,是累加操作,ai的值会多次加上,所以,从叶子节点开始操作是最优的,

所以我们进行拓扑排序,每次把入度为0的点存入,并且如果ai >= 0让他的下一个节点+ai

对于叶子节点ai < 0的时候,累计会导致多次加上一个负数,肯定是不行的,所以我们

遍历到入度为0,的点并且ai<0,把它存到栈中

拓扑排序完拿的话,就变成了,先拿-3,然后-2,-1明显这是最优情况 

#include 
#include 
#include 
#include
#include
#include
#include