leetcode: Best Time to Buy and Sell Stock

140 篇文章 0 订阅

Say you have an array for which the ith element is the price of a given stock on day i.

If you were only permitted to complete at most one transaction (ie, buy one and sell one share of the stock), design an algorithm to find the maximum profit.

比原来少一次循环

class Solution {
public:
    int maxProfit(vector<int> &prices) {
        if( prices.size() < 2)
            return 0;
        vector< int> dp( prices.size(), INT_MAX);
        int max_profit = INT_MIN;
        dp[0] = prices[0];
        for( int i = 1; i < prices.size(); ++i){
            dp[i] = min( dp[i-1], prices[i]);
            max_profit = max_profit >  prices[i] - dp[i] ? max_profit : prices[i] - dp[i];
        }
        return max_profit;
    }
};



一维动态规划:dp[i]保存当前下标之前的最低股价

dp[i] = min( dp[i-1], prices[i]);

代码中注释掉部分的解法会TLE,后来发现二了,只要比较dp[i-1]和当前值的大小就够了,速度快很多,时间复杂度从O(n2)变为O(n)

class Solution {
public:
    int maxProfit(vector<int> &prices) {
        if( prices.size() < 2)
            return 0;
        vector< int> dp( prices.size(), INT_MAX);
        dp[0] = prices[0];
        for( int i = 1; i < prices.size(); ++i){
            //int min = INT_MAX;
            //for( int j = i; j >= 0; --j){
            //    min = min < prices[j] ? min : prices[j];
            //}
            dp[i] = min( dp[i-1], prices[i]);
            //dp[i] = min;
        }
        int max_profit = INT_MIN;
        for( int i = 0; i < prices.size(); ++i)
            max_profit = max_profit > ( prices[i] - dp[i]) ? max_profit : prices[i] - dp[i];
        return max_profit;
    }
};


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值