股票问题(分治)

股票买卖 查看 提交 统计 提问 总时间限制: 1000ms 内存限制: 65536kB 描述

最近越来越多的人都投身股市,阿福也有点心动了。谨记着“股市有风险,入市需谨慎”,阿福决定先来研究一下简化版的股票买卖问题。

假设阿福已经准确预测出了某只股票在未来 N
天的价格,他希望买卖两次,使得获得的利润最高。为了计算简单起见,利润的计算方式为卖出的价格减去买入的价格。

同一天可以进行多次买卖。但是在第一次买入之后,必须要先卖出,然后才可以第二次买入。

现在,阿福想知道他最多可以获得多少利润。

输入 输入的第一行是一个整数 T (T <= 50) ,表示一共有 T 组数据。 接下来的每组数据,第一行是一个整数 N (1 <= N
<= 100, 000) ,表示一共有 N 天。第二行是 N 个被空格分开的整数,表示每天该股票的价格。该股票每天的价格的绝对值均不会超过
1,000,000 。 输出 对于每组数据,输出一行。该行包含一个整数,表示阿福能够获得的最大的利润。 样例输入 3 7 5 14 -2 4
9 3 17 6 6 8 7 4 1 -2 4 18 9 5 2 样例输出 28 2 0 提示 对于第一组样例,阿福可以第 1 次在第 1
天买入(价格为 5 ),然后在第 2 天卖出(价格为 14 )。第 2 次在第 3 天买入(价格为 -2 ),然后在第 7 天卖出(价格为
17 )。一共获得的利润是 (14 - 5) + (17 - (-2)) = 28 对于第二组样例,阿福可以第 1 次在第 1
天买入(价格为 6 ),然后在第 2 天卖出(价格为 8 )。第 2 次仍然在第 2 天买入,然后在第 2 天卖出。一共获得的利润是 8 -
6 = 2 对于第三组样例,由于价格一直在下跌,阿福可以随便选择一天买入之后迅速卖出。获得的最大利润为 0

#include <iostream>
#include <stdio.h>
#include <algorithm>
#include<cstring>
using namespace std;
//const int maxn=1005;
int a[100005];
int pre[100005];
int pro[100005];
int mina;
int maxa;
int main()
{
    int num;
    scanf("%d",&num);
    while(num--)
    {
        memset(pre, 0, sizeof(pre));
        memset(pro, 0, sizeof(pro));
        int N;
        scanf("%d",&N);
        for(int i=0;i<N;++i)
            scanf("%d",&a[i]);
        mina=a[0];
        for(int i=1;i<N;++i)
        {
            mina=min(mina,a[i]);
            pre[i]=max(pre[i-1],a[i]-mina);
        }//前i天的最大收益为前一天的最大收益和第i天卖(前i天最最便宜的);
        maxa=a[N-1];
        for(int i=N-1;i>=0;--i)
        {
            maxa=max(maxa,a[i]);
            pro[i]=max(pro[i-1],maxa-a[i]);
        }////后n-i+1天买卖的最大收益为后n-(i-1)+1天买卖的最大收益与第i天买入,后n-i+1天中最大价格卖出的利润中的较大值
        int ans=0;
        for(int i=0;i<N;++i)
            ans=max(ans,pre[i]+pro[i]);
        cout<<ans<<endl;
    }
}

猜你喜欢

转载自blog.csdn.net/Liusyu6688/article/details/81215049
今日推荐