0
点赞
收藏
分享

微信扫一扫

(每日一练c++)CC28 买卖股票的最好时机 iii

木匠0819 2022-05-02 阅读 30
c++

描述

假设你有一个数组,其中第i个元素是某只股票在第i天的价格。
设计一个算法来求最大的利润。你最多可以进行两次交易。
注意:
你不能同时进行多个交易(即,你必须在再次购买之前出售之前买的股票)。

示例1

输入:

[1,4,2]

返回值:

3

示例2

输入:

[2,4,1]

返回值:

2
class Solution {
public:
    int maxProfit(vector<int> &prices) {
        //You may complete at most two transactions.(you must sell the stock before you buy again)
        //方式一:爆搜 (630ms 8552k)
        /*
        int len=prices.size();
        int maxPro=0;
        for(int i=0;i<len;i++){ //以i作为两次交易的分界线
            int max1=0,max2=0;
            for(int j=0;j<=i;j++){
                for(int k=j+1;k<=i;k++){
                    int val = prices[k]-prices[j];
                    if(val>max1){
                        max1=val;
                    }
                }
            }
            for(int j=i;j<len;j++){
                for(int k=j+1;k<len;k++){
                    int val = prices[k]-prices[j];
                    if(val>max2){
                        max2=val;
                    }
                }
            }
            if(max1+max2>maxPro)maxPro = max1+max2;
        }
        return maxPro;
        */
        
        
        //方式二:波峰波谷检测法,可以减小不必要的搜索空间,相比于第一种效率提升了不少    (180ms 8568k)
        int len = prices.size();
        if(len<=1)return 0;
        vector<int> flag(len,0);
        for(int i=1;i<len-1;i++){
            if(prices[i]>prices[i-1]&&prices[i]>=prices[i+1])flag[i]=1;  //标记为波峰(比较时注意相等情况)
            if(prices[i]<prices[i-1]&&prices[i]<=prices[i+1])flag[i]=-1;  //标记为波谷
        }
        if(prices[0]<=prices[1])flag[0]=-1;
        if(prices[len-1]>prices[len-2])flag[len-1]=1;
        //搜索
        int maxPro=0;
        for(int k=0;k<len;k++){
            int max1=0,max2=0;
            for(int i=0;i<=k;i++){
                if(flag[i]==-1){
                    for(int j=i+1;j<=k;j++){
                        if(flag[j]==1){
                            int val = prices[j]-prices[i];
                            if(val>max1)max1 = val;
                        }
                    }
                }
            }
            for(int i=k;i<len;i++){
                if(flag[i]==-1){
                    for(int j=i+1;j<len;j++){
                        if(flag[j]==1){
                            int val = prices[j]-prices[i];
                            if(val>max2)max2 = val;
                        }
                    }
                }
            }
            int val = max1+max2;
            if(val>maxPro)maxPro = val;
        }
        return maxPro;
    }
};
举报

相关推荐

0 条评论