Leetcode 3530. Maximum Profit from Valid Topological Order in DAG

1. 解题思路

这一题的整体思路就是一个动态规划的思路,我们只需要在当前可以访问的位置当中遍历一下所有的可能性,然后取出其中的最大值即可。

因此,这里的核心问题也就变成了如何快速判断当前可以访问的位置。而这个是一个拓扑图的问题,要想要访问一个节点,其必须要求其依赖的节点都必须被访问过。因此,我们需要先根据给定的依赖关系确定一下每一个节点的依赖节点,然后我们只需要记录下当前所有已经访问过的节点,然后依次判断各个节点的依赖条件是否都被满足即可判断当前节点是否可以被取用。

另外,如果整张图没有任何节点,则我们可以简化问题,直接将其score排序之后依次访问即可。

2. 代码实现

给出python代码实现如下:

class Solution:
    def maxProfit(self, n: int, edges: List[List[int]], score: List[int]) -> int:
        if len(edges) == 0:
            return sum((i+1) * x for i, x in enumerate(sorted(score)))
        
        need = defaultdict(int)
        for u, v in edges:
            need[v]  = need[v] | (1 << u)
                 
        @lru_cache(None)
        def dp(idx, visited):
            if idx > n:
                return 0
            ans = 0
            state = 1
            for i in range(n):
                if (visited & state == 0) and (need[i] & visited == need[i]):
                    ans = max(ans, idx * score[i] + dp(idx+1, visited | state))     
                state = state << 1
            return ans
        
        return dp(1, 0)

提交代码评测得到:耗时214ms,占用内存24.3MB。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值