Content
121. Best Time to Buy and Sell Stock(E)
Description
Analysis & Solution
对prices遍历一次,用minBuy表示当前时间之前买入的最低价,maxProfit表示当前时间的最大利润。每次minBuy更新为当前price和minBuy的最小值,maxProfit更新为当前maxProfit和今天卖出最大利润(price-minBuy)的最大值。
Code
int maxProfit(vector<int>& prices)
{
if(prices.size() <= 1) return 0;
int minBuy = prices[0], maxProfit = 0;
for(int i = 1; i < prices.size(); i++)
{
minBuy = min(minBuy, prices[i]);
maxProfit = max(maxProfit, prices[i] - minBuy);
}
return maxProfit;
}
时间复杂度:遍历一次, O ( n ) O(n) O(n)。
122. Best Time to Buy and Sell Stock II(E)
Description
Analysis & Solution
此题目是上一题的扩展,可以在限制内进行任意多次交易,可用动态规划解决。
最优子结构dp[i][0]表示,当前持有股票时已获最大利润;dp[i][1]表示当前不持有股票时已获得的最大利润。
状态转移方程:
dp[i][0] = max(dp[i-1][0], dp[i-1][1] + prices[i])
dp[i][1] = max(dp[i-1][1], dp[i-1][0] - prices[i])
Code
int maxProfitII(vector<int>& prices)
{
int len = prices.size();
if(len <= 1) return 0;
vector<vector<int>> dp(len, vector<int> (2, 0));
dp[0][1] = - prices[0];
for(int i = 1; i < len; i++)
{
dp[i][0] = max(dp[i-1][0], dp[i-1][1] + prices[i]);
dp[i][1] = max(dp[i-1][1], dp[i-1][0] - prices[i]);
}
return dp[len-1][0];
}
时间复杂度:一次循环,时间复杂度为 O ( n ) O(n) O(n)。
124. Binary Tree Maximum Path Sum(H)
Description
Analysis & Solution
求树的最大路径和,对于某一结点node,包含该节点的最大路径和,有以下三种情况:
(1)路径不包含node的父节点,或node本身是根节点,最大和为node.val+dfs(node.left)+dfs(node.right)。
(2)路径包含node父节点,node.val+dfs(node.left)。
(3)路径包含node父节点,node.val+dfs(node.right)。
注意:dfs返回的值不包含第一种情况,所以返回值为node.val + max(leftSum, rightSum)。最大路径和maxSum比较当前值和上面三种情况的最大值进行更新。
Code
int maxSum = INT_MIN;
int maxPathSum(TreeNode *root)
{
int val = INT_MIN;
dfs(root);
return maxSum;
}
int dfs(TreeNode *root)
{
if(root == nullptr) return 0;
int leftSum = max(0, dfs(root->left));
int rightSum = max(0, dfs(root->right));
maxSum = max(maxSum, root->val + leftSum + rightSum);
return root->val + max(leftSum, rightSum);
}