Leetcode 045 Jump Game II(DP)

xiaoxiao2021-02-28  39

题目连接:Leetcode 045 Jump Game II

解题思路:动态规划,dp[i]表示到第i个位置最少需要几步。用一个优先队列维护在第i个位置之前最小的dp[k]值,每次取出一个最小的k,判断k+num[k]是否大于等于i,如果大于等于,那么dp[i] = dp[k] + 1;否则删除k,并再取出优先队列的头,直到dp[i]被更新,然后将dp[i]也放入优先队列中。

class Solution { public: int jump(vector<int>& nums) { int n = nums.size(); if (n == 0) return 0; int* dp = new int[n]; priority_queue<pair<int,int>> que; dp[0] = 0; que.push(make_pair(0, nums[0])); for (int i = 1; i < n; i++) { while (!que.empty() && que.top().second < i) que.pop(); dp[i] = 1 - que.top().first; que.push(make_pair(-dp[i], i+nums[i])); } int ans = dp[n-1]; delete dp; return ans; } };
转载请注明原文地址: https://www.6miu.com/read-2621134.html

最新回复(0)